R，对数据框的每隔一列应用函数？

Question

如何在数据框的每一列中应用一个函数？也就是说，如何修改df2 <- sapply(df1, fun) 使得df2 等于df1 但每隔一列应用一次fun？这是我尝试过的：

a <- c(1,2,3,4,5)
b <- c(6,7,8,9,10)
df1 <- data.frame(a,b)
df2 <- sapply(df1[c(TRUE, FALSE)], function(x) x^2)
isTRUE(dim(df1)==dim(df2)) # FALSE

此代码的问题在于，它删除了所有未应用 fun 的列 (dim(df2) # 5 1)。

Answer 1

将变量分配给切片

您可以为对象的子集分配新值。说：

x <- c(1,2,3)
x[2] <- 4

现在 x 将是 c(1,4,2)。同样，您可以对矩阵或数据框的行/列执行此操作。这里我们使用 apply 函数，第二个参数 2 表示 cols（1 表示 cols）。我推荐使用seq 函数来生成索引序列，from=1, by=2 给出奇数，from=2, by=2 给出偶数索引。指定这种方式可以推广到其他子集，并且可以直接检查您是否正确。

a <- c(1,2,3,4,5)
b <- c(6,7,8,9,10)
df1 <- data.frame(a,b)
df2 <- df1
df2[,seq(1, ncol(df2), 2)] <- apply(df2[,seq(1, ncol(df2), 2)], 2, function(x) x^2)

循环

请注意，您也可以使用循环来执行此操作：

df2 <- df1
for(col in seq(1, ncol(df2), 2)) df2[,col] <- sapply(df2[,col], function(x) x^2)

向量化函数

由于平方运算在 R 中是“矢量化”的，在这种情况下你也可以这样做：

for(col in seq(1, ncol(df2), 2)) df2[,col] <- df2[,col]x^2

或者完全使用矢量化：

df2 <- df1
df2[,seq(1, ncol(df2), 2)] <- df2[,seq(1, ncol(df2), 2)]^2