Python Pandas：旋转后重命名列

Question

我有一个旋转的熊猫数据框。我不知道如何更改列名，以便我可以以自然的方式继续使用透视数据框。下面是一个最小的工作示例。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非常感谢！

Answer 1

将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