使用字符串识别R中的列名序列

Question

我目前正在尝试使用预定义的字符串来识别R中的多个列名。为了更明确，我使用ave函数为数据帧的子组创建标识变量。扭曲是我希望识别变量是灵活的，这样我就可以将它作为通用字符串传递。

示例代码如下：

ids = with(df,ave(rep(1,nrow(df)),subcolumn1,subcolumn2,subcolumn3,FUN=seq_along))

我想以下面的方式运行此代码（下面的代码不能按预期工作）：

subColumnsString = c("subcolumn1","subcolumn2","subcolumn3")
ids = with(df,ave(rep(1,nrow(df)),subColumnsString ,FUN=seq_along))

我尝试使用eval，但仍然无法正常工作：

subColumnsString = c("subcolumn1","subcolumn2","subcolumn3")
ids = with(df,ave(rep(1,nrow(df)),eval(parse(text=subColumnsString)),FUN=seq_along))

有任何想法吗？谢谢。

编辑：我想要的工作代码示例：

df = mtcars
id_names = c("vs","am")
idDF_correct = transform(df,idItem = as.numeric(interaction(vs,am)))
idDF_wrong = cbind(df,ave(rep(1,nrow(df)),df[id_names],FUN=seq_along))

请注意，在idDF_correct中，唯一组合正确映射到idItem的唯一值。在idDF_wrong中，情况并非如此。

Answer 1

我认为这可以实现您的要求。在这里，我使用R附带的mtcars数据集：

subColumnsString <- c("cyl","gear")

ids = with(mtcars, ave(rep(1,nrow(mtcars)), mtcars[subColumnsString], FUN=seq_along))

只需使用子列索引data.frame，这些子列返回一个自然与ave一起使用的列表

编辑

ids = ave(rep(1,nrow(mtcars)), mtcars[subColumnsString], FUN=seq_along)

你可以省略with，只需要调用普通的'ol ave，就像G.Grothendieck所说的那样，你也应该使用他们的答案，因为它更通用。

Answer 2

这定义了一个函数，其参数为：

• data，输入数据框
• by，data中列名的字符向量
• fun，在ave中使用的功能

码 -

Ave <- function(data, by, fun = seq_along) {
   do.call(function(...) ave(rep(1, nrow(data)), ..., FUN = fun), data[by])
}

# test 
Ave(CO2, c("Plant", "Treatment"), seq_along)

赠送：

 [1] 1 2 3 4 5 6 7 1 2 3 4 5 6 7 1 2 3 4 5 6 7 1 2 3 4 5 6 7 1 2 3 4 5 6 7 1 2 3
[39] 4 5 6 7 1 2 3 4 5 6 7 1 2 3 4 5 6 7 1 2 3 4 5 6 7 1 2 3 4 5 6 7 1 2 3 4 5 6
[77] 7 1 2 3 4 5 6 7