循环转换应用/plyr

Question

我编写了以下代码，并希望通过使用plyr或apply来改进它，唉，我似乎无法让它工作。其思想是识别正确的列，然后提取/获取适当的值。

df <- data.frame(ID=c(1,2,3,4,5),COUNTRY=c('NL','FR','NL','BE','NL'),
             NL_GEN=c(5,5,10,20,5),FR_GEN=c(15,10,10,25,5),BE_GEN=c(10,10,5,15,5))

col_sel <- vector('numeric',length=nrow(df))
for ( i in 1:nrow(df)) {
  col_sel[i] <- as.numeric(df[i,grep(paste(df$COUNTRY[i],"_GEN",sep = ""),names(df))])
}

如何使用plyr或应用程序一次通过1行并返回单个值而不是整个列？

谢谢你能提供的任何帮助。

Answer 1

考虑vapply()和seq_along()，它允许您指定输出，这里是一个数字向量：

new_col_sel <- vapply(seq_along(df), function(i) {
      df[i, grep(paste0(df$COUNTRY[i], "_GEN"), names(df))]
}, numeric(1))

new_col_sel   
# [1]  5 10 10 15  5

all.equal(col_sel, new_col_sel)
# [1] TRUE 

如果需要向dataframe添加列，只需分配它：

df$col_sel <- vapply(seq_along(df), function(i) {
     df[i, grep(paste0(df$COUNTRY[i], "_GEN"), names(df))]
}, numeric(1))

#   ID COUNTRY NL_GEN FR_GEN BE_GEN col_sel
# 1  1      NL      5     15     10       5
# 2  2      FR      5     10     10      10
# 3  3      NL     10     10      5      10
# 4  4      BE     20     25     15      15
# 5  5      NL      5      5      5       5

Answer 2

我认为我们可以从adply包中使用plyr：

library(plyr)
adply(df, 1, function(x) c(Result = x[,which(grepl(x$COUNTRY, names(x)))]))

#   ID COUNTRY NL_GEN FR_GEN BE_GEN Result
# 1  1      NL      5     15     10      5
# 2  2      FR      5     10     10     10
# 3  3      NL     10     10      5     10
# 4  4      BE     20     25     15     15
# 5  5      NL      5      5      5      5