r - 使用 R 将函数应用于数据框中两列的每个组合

Question

我想将函数 myfoo 应用于数据帧 mydf 中两列的每种可能组合，并以矩阵格式获取结果 myoutput.

考虑以下数据框，

# Example dataframe
mydf <- data.frame(var1 = 1:10, var2 = 11:20, var3 = 21:30)
head(mydf)

# var1 var2 var3
# 1    11   21
# 2    12   22
# 3    13   23
# 4    14   24
# 5    15   25

我想将以下函数应用于两列的每种可能的组合，

# Example function
myfoo <- function(varA, varB) sum(varA * varB)

myfoo(var1, var2)
# [1] 935

为了获得此输出。

# Desired output
myoutput <- matrix(c(0, 935, 1485, 935, 0, 4035, 1485, 4035, 0), 3, dimnames = list(names(mydf), names(mydf)))
myoutput

#      var1 var2 var3
# var1    0  935 1485
# var2  935    0 4035
# var3 1485 4035    0

Answer 1

在你的情况下，我会转换为矩阵(当所有列都是数字类时，没有理由将其保留为data.frame)并运行编译后的crossprod进行矩阵叉积的函数。

m <- as.matrix(mydf)
res <- crossprod(m, m)
diag(res) <- 0 # You can probably skip that part
res
#      var1 var2 var3
# var1    0  935 1485
# var2  935    0 4035
# var3 1485 4035    0