在python数据框

Question

我在python数据框中有一个列，它可以是_00_test或00_test。我想重命名列进行测试。

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

怎么做？

Answer 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