我有一只熊猫。看起来像这样的DataFrame:
索引 | 项目id | 问题答案 | >|
---|---|---|---|
0 | 1 | 'q1' | 'tr1' |
1 | 1 | 'q2' | 'tr2' |
2 | 1 | 'q3' | 'tr3'|
3 | 2 | 'q1' | 'tr4' |
2 | 'q3' | 'tr6' |
您可以在旋转之前使用cumcount
来获取后缀:
df['idx'] = df.groupby('projectid').cumcount() + 1
df = df.pivot(index='projectid',columns='idx')[['question','answer']]
df.columns = [''.join(map(str, col)) for col in df.columns]
print(df)
输出::
question1 question2 question3 answer1 answer2 answer3
projectid
1 'q1' 'q2' 'q3' 'str1' 'str2' 'str3'
2 'q1' 'q3' NaN 'str4' 'str6' NaN