R中的apply（）具有用戶定義的函數

Question

我有一個數據框，其中有投票和政黨標簽

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")

誰能發現問題？ 相關，這可以更有效地實現嗎？

Answer 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"