我有一个数据框n
列并希望对每个列应用一个函数组合列。这与如何cor()
函数将数据帧作为输入并生成相关矩阵作为输出,例如:
X <- data.frame(A=rnorm(100), B=rnorm(100), C=rnorm(100))
cor(X)
这将生成以下输出:
> cor(X)
A B C
A 1.00000000 -0.01199511 0.02337429
B -0.01199511 1.00000000 0.07918920
C 0.02337429 0.07918920 1.00000000
但是,我有一个自定义函数,需要将其应用于每个列组合。我现在使用的解决方案使用嵌套 for 循环,该解决方案有效:
f <- function(x, y) sum((x+y)^2) # some placeholder function
out <- matrix(NA, ncol = ncol(X), nrow = ncol(X)) # pre-allocate
for(i in seq_along(X)) {
for(j in seq_along(X)) {
out[i, j] <- f(X[, i], X[, j]) # apply f() to each combination
}
}
其产生:
> out
[,1] [,2] [,3]
[1,] 422.4447 207.0833 211.4198
[2,] 207.0833 409.1242 218.2430
[3,] 211.4198 218.2430 397.5321
我目前正在尝试过渡到 tidyverse,并且希望避免使用 for 循环。有人可以给我看一个tidy这种情况的解决办法? Thanks!