Python：透视数据框以引入新列

Question

我有以下数据框

df = 
patient_id code
1          A
1          B
1          C
2          A
2          B
3          A

我想旋转它以获得以下内容：

patient_id code_1 code_2 code_3
1          A      B      C
2          A      B      NA
3          A      NA     NA

Answer 1

你可以pivot调用'code'两次：

df.pivot('patient_id', 'code', 'code')

这给了

code        A    B    C
patient_id
1           A    B    C
2           A    B  NaN
3           A  NaN  NaN

然后只需重命名您的列。