python - 循环更改数据框标题 - python

Question

我正在尝试遍历数据框的标题并将它们更改为标题的最后一个字符，以便我可以将它们转换为月份。我想要一个循环，因为我每个月都会获取数据并且我希望它自动化，所以我不手动添加标题名称。当我打印标题时，这就是我得到的：

list(df)

['MCA036_LIFETIME_1', 'MCA036_LIFETIME_2', 'MCA036_LIFETIME_3', 'Total']

我希望标题是：

list(df)

['January','February','March','Total']

任何指导将不胜感激。

Answer 1

您不需要循环，只需：

df.columns = ['January','February','March','Total']

---

如果您想自动转换该列表中的字符串，这是一个一般的 Python 问题，与 pandas 没有特别的关系：

In [141]: all_monthes = ['January', 'February', 'March']

In [142]: l = ['MCA036_LIFETIME_1', 'MCA036_LIFETIME_2', 'MCA036_LIFETIME_3', 'Total']

In [143]: [all_monthes[int(e.split('_')[-1])-1] if e != 'Total' else e for e in l]
Out[143]: ['January', 'February', 'March', 'Total']