Groupby 一个基于相同长度的 pandas.DataFrame 的 groupby 的 numpy.array

Question

我有一个 numpy.array arr 和一个 pandas.DataFrame df。

arr 和 df 具有相同的形状 (x,y)。

我需要按df 的一列分组，并在具有相同形状的arr 上应用受影响行的转换。

为了清楚起见，这是一个玩具示例：

arr = 
   0  1   12   3
   2  5   45   47
   3  19  11  111

df =
   A  B   C   D
0  0  1   2   3
1  4  5   6   7
2  4  9  10  11

我想将df 与A 分组并计算平均值，但不是转换df 我希望转换arr。

所以我得到类似的东西：

    arr = 
        0        1         12          3
       (2+3)/2  (5+19)/2   (45+11)/2   (47+111)/2

这可能吗？没有昂贵的循环？

提前致谢

Answer 1

看起来需要首先从arr 创建DataFrame，然后按列groupby 创建A 并聚合mean。最后由values 将其转换为numpy array：

print (pd.DataFrame(arr).groupby(df.A).mean().values)
[[  0.    1.   12.    3. ]
 [  2.5  12.   28.   79. ]]