我有一個旋轉的熊貓數據框。我不知道如何更改列名，以便我可以以自然的方式繼續使用透視數據框。下面是一個最小的工作示例。df = pd.DataFrame({'foo': ['one', 'one', 'one', 'two', 'two', 'two'], 'bar': ['A', 'B', 'C', 'A', 'B', 'C'], 'baz': [1, 2, 3, 4, 5, 6], 'zoo': ['x', 'y', 'z', 'q', 'w', 't']})旋轉后df.pivot(index='foo', columns='bar', values=['baz', 'zoo'])輸出是： baz zoobar A B C A B Cfooone 1 2 3 x y ztwo 4 5 6 q w t為了獲得以下輸出，以下步驟是什么？ A_baz B_baz C_baz A_zoo B_zoo C_zooone 1 2 3 x y ztwo 4 5 6 q w t非常感謝！
1 回答
犯罪嫌疑人X
TA貢獻2080條經驗 獲得超4個贊
將f-strings 與列表理解結合使用：
#python 3.6+
df.columns = [f'{j}_{i}' for i, j in df.columns]
#lower python versions
#df.columns = ['{}_{}'.format(j, i) for i, j in df.columns]
print (df)
A_baz B_baz C_baz A_zoo B_zoo C_zoo
foo
one 1 2 3 x y z
two 4 5 6 q w t
或DataFrame.swaplevel與map和join：
df = df.pivot(index='foo', columns='bar', values=['baz', 'zoo']).swaplevel(0,1,axis=1)
df.columns = df.columns.map('_'.join)
print (df)
A_baz B_baz C_baz A_zoo B_zoo C_zoo
foo
one 1 2 3 x y z
two 4 5 6 q w t
添加回答
舉報
0/150
提交
取消