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 的解决方案都要快。