【发布时间】:2018-05-08 14:11:53
【问题描述】:
我需要将 DataFrame 从长格式改成宽格式。数据外观示例:
import pandas as pd
df = pd.DataFrame({'id':[1,1,1,2,2,2,3,3], 'id_age':
30,30,30,23,23,23,29,29], 'product':['A','B','C','A','B','C','A','B'],
'rank':[1,2,3,3,1,2,2,1], 'result':['x','y','z','p','q','r','s','t']})
print(df)
id id_age product rank result
0 1 30 A 1 x
1 1 30 B 2 y
2 1 30 C 3 z
3 2 23 A 3 p
4 2 23 B 1 q
5 2 23 C 2 r
6 3 29 A 2 s
7 3 29 B 1 t
想要的输出是
id id_age product rank result
0 1 30 [A, B, C] [1, 2, 3] [x, y, z]
1 2 23 [A, B, C] [3, 1, 2] [p, q, r]
2 3 29 [A, B] [2, 1] [s, t]
即每个 id 一行。我试图通过创建一个数据透视表来解决这个问题,但我不知道如何:
- 为不止一列(产品、排名、结果)重塑它
- 处理个体中不变的值 (id_age)
- 使最终数据框中的列采用列表格式(顺序很重要,排名中列表的第一个值对应于给定个人结果中列表的第一个值)。
感谢任何有关如何在此问题上取得进展的建议!
【问题讨论】:
-
提醒一下,pandas 确实不支持在单元格中放置非标量条目。除非这是您流程的最后一步,否则您可能会遇到一些神秘的行为。
标签: python pandas pivot reshape