Question

我正在使用Python熊猫执行以下操作。我有一个F列和r行的数据框。此数据框中的所有字段均为数字。一列表示每一行所属的组。例如：

df = pd.DataFrame({'group': [1, 2, 1, 0, 0, 0, 1, 2],
            'A': [1.5, 0.5, 2.5, 0.5, 1.5, 0.5, 1.5, 0.5],
            'B': [3.5, 2.5, 3.5, 2.5, 3.5, 2.5, 3.5, 2.5]})

对于每个小组，我需要：

找到该组中每列的平均值，然后从每个元素中减去平均值

normalised_df = df.groupby(['group'], group_keys=True, axis=0).apply(lambda x: x - np.mean(x))

     group         A         B
0        1 -0.333333  0.000000
1        2  0.000000  0.000000
2        1  0.666667  0.000000
3        0 -0.333333 -0.333333
4        0  0.666667  0.666667
5        0 -0.333333 -0.333333
6        1 -0.333333  0.000000
7        2  0.000000  0.000000

对每个组进行转置并将其自身相乘（这样我就得到了每个组尺寸为FxF的矩阵）

例如第0组：

Group 0 transposed:
[-0.333333 0.666667 -0.333333
-0.333333 0.666667 -0.333333]

multiplied by itself:
[-0.333333 -0.333333
 0.666667  0.666667
-0.333333 -0.333333]

to give a 2x2 matrix

[0.666 0.666
 0.666 0.666]