python - 在 Python (Pandas-Numpy) 中，如何使用条件和迭代修改列名(字符串)？

Question

我正在尝试修改包含很多列的数据框的列名。列名是像这样的字符串:

'0000', '0005'...'0100'...'2355'

由于是大量的列，我需要通过迭代来做到这一点。修改的要点是，如果列名(字符串)以'0'开头，则修改该列名(字符串)，使新值仅为字符串的最后3位(所有字符串均为4位)。

所以我所做的是:

将列名放在列表中

 df_cols = df.columns.tolist()

然后通过迭代对列表进行修改

for i in range(len(df_cols)):
    if df_cols[i][0] == '0':
        df_cols[i] = df_cols[i][1:4]

当我检查列表时，它有效地进行了修改。但是，当我尝试在数据框中使用列名修改列表 (df_cols) 时:

df = df[df_cols]

我收到一条错误消息:

File "c:\users\hernan\anaconda\lib\site-packages\pandas\core\frame.py", line 1774, in __getitem__
return self._getitem_array(key)

File "c:\users\hernan\anaconda\lib\site-packages\pandas\core\frame.py", line 1818, in _getitem_array
indexer = self.ix._convert_to_indexer(key, axis=1)

File "c:\users\hernan\anaconda\lib\site-packages\pandas\core\indexing.py", line 1143, in _convert_to_indexer
raise KeyError('%s not in index' % objarr[mask])

KeyError: "['000' '001' '002' '003' '004' '005' '006' '007'....] not in index"

感谢帮助

Answer 1

您刚刚更改了 df_cols 的值。您必须先更新 DataFrame 的列名，然后才能使用它们:

df.columns = df_cols