如何根据与行索引对应的列表获取数据帧的子集？

Question

我有一个看起来像这样的 pandas 数据框：

row|a|b|
0 |1|2
1 |2|0
2 |1|1
3 |3|2
4 |1|1
5 |1|0

我想获取每列的平均值，但只针对第 1、3、4 行，然后与第 0、2、5 行的平均值进行比较。索引存储在一个 numpy 数组 keep = np.array([1,3,4]) 和 take = np.array([0,2,5]) 中。理想的输出应该是如下所示的数组或数据框：

keep |2|1
take |1|1

有人知道怎么做吗？

Answer 1

concat 怎么样：

pd.concat({'keep':df.iloc[keep].mean(), 
           'take':df.iloc[take].mean()}
         ).unstack()

输出：

        a    b
keep  2.0  1.0
take  1.0  1.0