在dataframe中对每3列(cols: 1-3、4-6、7-9)应用函数

Question

我有以下具有多索引的DataFrame：

在这里输入图像描述

在每一行中，对每3列应用平均值/和/avg函数的最佳方法是:a、b、c，然后是: a1 b1 c1和a2 b2 c2，因此结果是:在示例中，我做了sum 在这里输入图像描述

Answer 1

您可以通过groupby和聚合函数(如sum )使用numpy array。

df = pd.DataFrame({'A':[1,2,3],
                   'B':[4,5,6],
                   'C':[7,8,9],
                   'D':[1,3,5],
                   'E':[5,3,6],
                   'F':[7,4,3]})

print (df)
   A  B  C  D  E  F
0  1  4  7  1  5  7
1  2  5  8  3  3  4
2  3  6  9  5  6  3

print (np.arange(len(df.columns)) // 3)
[0 0 0 1 1 1]

print (df.groupby(np.arange(len(df.columns)) // 3, axis=1).sum())
    0   1
0  12  13
1  15  10
2  18  14