Python pandas - 对字段进行分组和汇总

时间:2015-05-22 22:25:23

标签: python pandas dataframe

我最近一直在玩Panda的DataFrames,并努力分析一些多维数据。

我们说我有一些数据如下:

order | sample | feature1 | feature2
-------------------------------------
1234  | A      | 0.20     | 0.45
1234  | B      | 0.71     | 0.08
1234  | C      | 0.21     | 0.02
1234  | D      | 0.87     | 0.88
5678  | A      | 0.76     | 0.42
5678  | B      | 0.01     | 0.03
5678  | C      | 0.29     | 0.91
5678  | D      | 0.70     | 0.78

我希望输出按顺序分组的所有内容以及每个功能按样本汇总的位置:

order | feature1                  | feature2 
      | A    | B    | C    | D    | A    | B    | C    | D   
------------------------------------------------------------
1234  | 0.20 | 0.71 | 0.21 | 0.87 | 0.45 | 0.08 | 0.02 | 0.88
5678  | 0.76 | 0.01 | 0.29 | 0.70 | 0.42 | 0.03 | 0.91 | 0.78

这是我到目前为止所做的:

from pandas import *
df = DataFrame({"order": [1234, 1234, 1234, 1234, 5678, 5678, 5678, 5678], "sample": ["A", "B", "C", "D", "A", "B", "C", "D"], "feature1": [0.20, 0.71, 0.21, 0.87, 0.76, 0.01, 0.29, 0.70], "feature2": [0.45, 0.08, 0.02, 0.88, 0.42, 0.03, 0.91, 0.78]})
byorder = df.groupby("order")
# not sure how to go from 1 groupby object to a new dataframe having what i need

您对我如何最终获得包含我需要的聚合数据的新DataFrame有什么想法吗?也许DataFrames不适合进行这种操作?

1 个答案:

答案 0 :(得分:6)

您可以使用pivot

>>> df.pivot(index='order', columns='sample')
       feature1                   feature2
sample        A     B     C     D        A     B     C     D
order
1234       0.20  0.71  0.21  0.87     0.45  0.08  0.02  0.88
5678       0.76  0.01  0.29  0.70     0.42  0.03  0.91  0.78