Question

我有一个像

这样的数据框

 X     Y    VALUE    
"AAA" "BBB"
"CCC" "AAA"

我还有一个矩阵：

     AAA BBB CCC
 AAA  1   13  2
 BBB  27  11  4
 CCC  6   12  3

使用AAA，BBB和CCC作为列名和行名。如何使用矩阵中的值填充数据框中的VALUE列？

Answer 1

你可以做到

df$VALUE <- mat[cbind(df$X, df$Y)]

使用您的示例：

df <- data.frame(X = c("AAA", "CCC"),
                 Y = c("BBB", "AAA"), stringsAsFactors = FALSE)
df
#     X   Y
# 1 AAA BBB
# 2 CCC AAA

mat <- matrix(c(1, 27, 6, 13, 11, 12, 2, 4, 3),
              nrow = 3, ncol = 3,
              dim = list(c("AAA", "BBB", "CCC"),
                         c("AAA", "BBB", "CCC")))
mat
#     AAA BBB CCC
# AAA   1  13   2
# BBB  27  11   4
# CCC   6  12   3

df$VALUE <- mat[cbind(df$X, df$Y)]
df
#     X   Y VALUE
# 1 AAA BBB    13
# 2 CCC AAA     6

Answer 2

您还可以使用melt：

直接从矩阵中获取数据帧

library(reshape2)
melt(mat)

根据数据框中的字符串在矩阵中查找值

2 个答案: