Question

我正在尝试修改包含大量列的数据框的列名。列名是字符串，如：

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

由于是大量的列，我需要通过迭代来完成此操作。修改的要点是，如果列名（字符串）以'0'开头，则修改该列名（字符串），以使新值仅为字符串的最后3位数（所有stings都有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

Answer 2

您正在修改列的副本，而不是实际的column_names。这应该做：

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.columns = df_cols  #Here you substitute back the modified column names to the dataframe

希望它有所帮助.. :)

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

2 个答案: