python - 更改链式操作中的所有列名

Question

我有一个数据框:

df = pd.DataFrame(data=[[1,2]], columns=['a', 'b'])

我知道我可以执行以下操作来更改数据框中的所有列名称:

df.columns = ['d', 'e']

如何更改链式操作中的所有列名称？例如，我想做这样的事情:

df=(
    df.rename all column names
    .reset_index()
)

我能找到的唯一方法是使用 df.rename 并用新旧列对构建一个字典，但这看起来非常难看。有没有更优雅的解决方案？

谢谢。

Answer 1

感谢@unutbu 指向 git hub issue ，事实证明这可以通过其中一条评论中的 set_axis 来完成:

df = pd.DataFrame(data=[[1,2]], columns=['a', 'b'])   
df
Out[21]: 
   a  b
0  1  2

df2 = (
    df.set_axis(['d','e'], axis=1, inplace=False)
    .reset_index()
)
df2
Out[18]: 
   index  d  e
0      0  1  2