如何将数据框列值转换为没有循环的数组

Question

我有一个这样的数据框：

df = pd.DataFrame({'A': [10,10,11,14], 'B':[2,3,3,5]})

看起来像这样：

    A   B
0   10  2
1   10  3
2   11  3
3   14  5

我想转换为此，使用A作为行索引，并将B的值存储在数组或矩阵中：

10   2  3
11   3
14   5

有没有python方式这样做而不查看数据框df中的每一行？

非常感谢

Answer 1

使用groupby：

df.groupby('A')

然后你可以（例如）通过以下方式获得分组版本的平均值：

df.groupby('A').mean()

导致：

      B
A      
10  2.5
11  3.0
14  5.0