在python数据框中根据条件重命名列

【问题标题】：Rename columns on condition in python data frame在python数据框中根据条件重命名列
【发布时间】：2017-03-15 21:46:36
【问题描述】：

我在 python 数据框中有一列，它将是 _00_test 或 00_test 。我想重命名要测试的列。

如果存在_00_test 或00_test，我要做的是检查数据框中的列，然后将列名重命名为test

怎么做？

【问题讨论】：

标签： python dataframe pyspark

【解决方案1】：

DataFrame.rename 如果列不在数据框中，则不会失败。因此，如果我们假设排他性，数据框中只能存在一列，那么您可以一次性完成：：

>>> import pandas as pd
>>> df = pd.DataFrame({'_00_test': range(5)})
>>> df
   _00_test
0         0
1         1
2         2
3         3
4         4
>>> df.rename(columns={'_00_test': 'test', '00_test': 'test'})
   test
0     0
1     1
2     2
3     3
4     4

【讨论】：