我想知道是否有办法直接从 apply
返回数据帧或 plyr
当函数返回的列数可变(但行数始终相同)时调用。例如:
df <- data.frame(A = 1:3, B = c("a","b", "c"))
my_fun <- function(x){
if(is.numeric(unlist(x))){
return(x)
} else {
return(cbind(x, x))
}
}
我能够得到的最接近的是返回一个列表并将其转换为数据框:
library(plyr)
data.frame(alply(df, 2, my_fun))
## A X2.B X2.B.1
## 1 1 a a
## 2 2 b b
## 3 3 c c
感觉应该有一种方法可以在没有额外转换的情况下做到这一点,是吗?
最佳答案
我用 lapply()
当您想将函数应用于数据框的多列时,以这种方式很多。在基础 R 中,您可以将数据框视为列表,其中每一列是一个元素。如果你像往常一样使用 lapply() 它将返回一个列表,这不是我们想要的。
> lapply(df, my_fun)
$A
[1] 1 2 3
$B
x x
[1,] 1 1
[2,] 2 2
[3,] 3 3
但是如果你把结果赋值给
df[]
它会向 R 发出信号,表明您想要原始数据框的子集(完整子集,根本不是子集),从而保留数据框对象类型。> df[] <- lapply(df, my_fun)
> df
A B.x B.x
1 1 1 1
2 2 2 2
3 3 3 3
关于r - 当返回的列数可变时使用 apply 或 plyr,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35077420/