使用一串列名对数据框进行子集化

Question

我需要通过我创建的一串列名称对数据框 (df) 进行子集化 - 不知道如何将其注入子集..？

例如

colstoKeep is a character string:
"col1", "col2", "col3", "col4"

如何将其推入子集函数

df<- df[colstoKeep]

我确定这很容易。？因为上面的行不通。

Answer 1

df  <- data.frame(A=seq(1:5),B=seq(5:1),C=seq(1:5))
df

colsToKeep <- "\"A\", \"C\""

如果我正确理解您的问题，您的 colsToKeep 变量是上面给出的字符串。为了提取变量，您必须将其转换为向量。如果我使用了正确的格式，您可以使用以下代码来做到这一点。

library(magrittr)
colsToKeepVector <- 
  strsplit(colsToKeep, ",") %>%
  unlist() %>%
  trimws() %>%
  gsub("\"", "", .)

df[colsToKeepVector]

但是，如果我也了解您有一个向量并折叠为字符串 (paste(..., collapse = ", ")?)，我强烈建议您不要这样做。

（已编辑以匹配问题中的字符串格式）

Answer 2

df  <- data.frame(A=seq(1:5),B=seq(5:1),C=seq(1:5))
df
  A B C
1 1 1 1
2 2 2 2
3 3 3 3
4 4 4 4
5 5 5 5

cols_to_keep  <- c("A","C")
df[,cols_to_keep]
  A C
1 1 1
2 2 2
3 3 3
4 4 4
5 5 5