循环遍历列表时出错：“`[<-.data.frame`(`*tmp*`, , i, value = c(7L, 1L, 4L, 7L, 7L, : 新列会留下漏洞... "

Question

我正在尝试编写一个循环遍历列表的函数，以便仅在数据集的特定列上运行 kmeans 聚类。当在每组列上运行 kmeans 时，我希望输出是每个观察值的集群成员的矩阵/数据框。

这是一个模拟数据集和我想出的函数（我是 R 新手——对不起，如果它不稳定）

set.seed(123)
mydata <- data.frame(a = rnorm(100,0,1), b = rnorm(100,0,1), c = 
rnorm(100,0,1), d = rnorm(100,0,1), e = rnorm(100,0,1)) 

set.seed(123)
my.kmeans <- function(data,k,...) {
    clusters <- data.frame(matrix(nrow = nrow(data), ncol = 
    length(list(...)))) # set up dataframe for clusters
    for(i in list(...)) {
        kmeans <- kmeans(data[,i],centers = k)
        clusters[,i] <- kmeans$cluster
    }
    colnames(clusters) <- list(...)
    clusters
}

我的问题是：当我只要求它使用连续的列时，这似乎有效，但当我要求它跳过一些列时却不行。例如，以下第一个有效，但第二个无效。知道如何解决这个问题吗？

# works how I want 
head(my.kmeans(data = mydata, k = 8, c(1,2), c(2,3), c(1,2,3)))

# doesn't work 
head(my.kmeans(data = mydata, k = 8, c(1,2), c(2,3), c(1,2,5)))

另外，我知道人们建议使用 apply 函数并远离 for 循环，但我不知道如何使用 apply 函数来做到这一点。对此的任何建议也将不胜感激。

非常感谢！

丹尼

Answer 1

以@SatZ 的 cmets 为基础，

set.seed(123)
mydata <- data.frame(a = rnorm(100,0,1), b = rnorm(100,0,1), c = 
                   rnorm(100,0,1), d = rnorm(100,0,1), e = 
                   rnorm(100,0,1)) 
mylist <- list(c(1,2), c(2,3), c(1,2,5))

set.seed(123)
my.kmeans <- function(data,k,list) {
  clusters <- data.frame(matrix(nrow = nrow(data), ncol = 
                              length(list))) # set up dataframe for 
                              clusters
  for(i in 1:length(list)) {
      kmeans <- kmeans(data[,list[[i]]],centers = k)
      clusters[,i] <- kmeans$cluster
  }
  colnames(clusters) <- list
  clusters
}

head(my.kmeans(data = mydata, k = 8, list = mylist))