使用用户定义函数在 R 中应用（）答案

【问题标题】：apply() in R with user-defined function使用用户定义函数在 R 中应用（）
【发布时间】：2013-12-19 00:12:59
【问题描述】：

我有一个数据框，其中包含如此排列的选票和政党标签

dat <- data.frame( v1=c(25, 0, 70), 
                   v2=c(75, 100, 20), 
                   v3=c(0, 0, 10), 
                   l1=c("pA", ".", "pB"), 
                   l2=c("pB", "pC", "pC"), 
                   l3=c(".", ".", "pD") )

这样每一行都是一个分析单元。只有得票方需要考虑，这个函数提取正票或相应的标签

getpos <- function(vector, vorl="v"){ # change to "l" to report labels
    vot <- vector[grep( "v", colnames(vector) )]; 
    lab <- vector[grep( "l", colnames(vector) )];
    if (vorl=="v") {vot[vot>0]} else {lab[vot>0]};
}
getpos(dat[1,])           # votes for obs 1
getpos(dat[1,], vorl="l") # labels for obs 1

我希望在数据帧 dat 的每一行中运行函数 getpos 以生成具有不同长度的投票/标签向量的列表。应用该函数并没有返回我所期望的：

apply(X=dat, MARGIN=1, FUN=getpos, vorl="l")

谁能发现问题？和相关的，这可以更有效地实现吗？

【问题讨论】：

apply 所做的第一件事是将dat 转换为矩阵，这将导致您的所有数字都转换为字符。您需要从根本上重新考虑如何组织这些数据。可能是长而不是宽格式。
其次@joran 的评论 - 如果您重组为 data.frame(number=c(1,1,2,3,3,3), party=c("pA","pB","pC","pB","pC","pD"), votes=c(25,75,100,70,20,10)) 之类的东西，您将更适合使用 aggregate 函数等。
@joran 的组织建议看起来很有希望，我会尝试并报告。谢谢！

标签： r apply

【解决方案1】：

这里发生的情况是数据框中的行在被apply 提取后不再具有列名（但它们确实有names）：

试试：

getpos <- function(x, vorl="v"){ 
     vot <- x[grep( "v", names(x) )] ;  lab <- x[grep( "l", names(x) )];
     if (vorl=="v") {vot[vot>0]} else {lab[vot>0]};
 }

> apply(dat, MARGIN=1, FUN=function(x2) getpos(x2, vorl="l") )
#-------------
[[1]]
  l1 
"pA" 

[[2]]
  l2 
"pC" 

[[3]]
  l1   l3 
"pB" "pD"

【讨论】：

感谢 @DWin 解释 apply 的作用。您的解决方案似乎工作正常。感谢您分享知识！