如何基于与行索引相对应的列表来获取数据帧的子集

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数组CCD_ 1和CCD_。理想的输出是一个数组或数据帧，看起来如下：

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