是否有一个 R 函数可以从数据框中选择 N 个随机列？

Question

是否有一个 R 函数可以从数据框中选择 N 个随机列？我正在尝试检查用于贝叶斯网络结构学习的 Sparsebn 包的时间复杂度

我已经尝试过了，但是该算法不仅选择 N 列，而且还选择 N 行。如何解决这个问题？

library(sparsebn)
library(igraph)
library(graph)

df <- read.csv("data/arth150.csv", header = TRUE, sep = ",", check.names = FALSE)

df <- as.data.frame(unclass(df), stringsAsFactors = TRUE)

experiment_range <- list(10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 106)

timelist <- list()

for (i in experiment_range) {
  rand_df <- df[sample(ncol(df), size=i), ]
  start_time <- Sys.time()
  dat <- sparsebnData(rand_df, type = 'c')
  dags <- estimate.dag(data = dat)
  end_time <- Sys.time()
  ctime <- end_time - start_time
  otime <- list(ctime)
  timelist <- append(timelist, otime)
}

Answer 1

如果 df 是一个数据帧，您可以通过执行以下操作随机采样 i 列：

df[,sample(1:ncol(df),i)]

Answer 2

或者使用 dplyr：

dplyr::select(df, sample(seq_len(ncol(df)), size = i))

在管道中：

df %>% dplyr::select(sample(seq_len(ncol(.)), size = i))