R数据帧按列名连接

Question

当我尝试加入2个数据框时，我有一个相当独特的问题（我相信），其中连接条件是列名（不是值）。让我用一个例子解释一下数据： 这是我的预测数据框的头部（多类预测）：

> head(mnm.predict.test.probs)
              1            2          3
9  1.013755e-04 3.713862e-02 0.96276001
10 1.904435e-11 3.153587e-02 0.96846413
12 6.445101e-23 1.119782e-11 1.00000000
13 1.238355e-04 2.882145e-02 0.97105472
22 9.027254e-01 7.259787e-07 0.09727389
26 1.365667e-01 4.034372e-01 0.45999610

这是响应数据帧的负责人：

> head(testing.logist$cut.rank)
[1] 3 3 3 3 1 3

这两组之间的连接应该在第一个数据帧中查找第二个数据帧中相应值的概率。例如： 返回的数据框/列表应如下所示：

0.96276001
0.96846413
1.00000000
0.97105472
9.027254e-01
0.45999610

知道如何有效地做到这一点吗？

Answer 1

[子集运算符还接受一个矩阵作为参数，其中每一行代表您想要获取的元素的行索引和列索引。试试这个：

mnm.predict.test.probs[cbind(1:nrow(mnm.predict.test.probs),testing.logist$cut.‌​‌​rank)]
#[1] 0.9627600 0.9684641 1.0000000 0.9710547 0.9027254 0.4599961

作为内部运算符，这比任何for循环或基于*apply的解决方案都要快。