我最近一直在玩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不适合进行这种操作?
答案 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