如何在Python（Pandas）中重命名特定范圍的列

Question

 KOAK.rename(columns = lambda x : 'KOAK-' + x, inplace=True)

上面的代碼允許我在數據框中所有列標題的前面添加“ KOAK”，但是我只想將KOAK前綴添加到第4-38列。 我將如何去做呢？

Answer 1

您可以分配給columns ：

KOAK.columns = ['KOAK-' + x if 4 <= i <= 38 else x for i, x in enumerate(KOAK.columns, 1)]

例：

KOAK = pd.DataFrame(dict.fromkeys('ABCDEFG', [12]))
KOAK.columns = ['KOAK-' + x if 2 <= i <= 4 else x 
                for i, x in enumerate(KOAK.columns, 1)]
print(KOAK)

打印：

    A  KOAK-B  KOAK-C  KOAK-D   E   F   G
0  12      12      12      12  12  12  12

Answer 2

您可以通過使用列表推導壓縮列序范圍並將其傳遞給rename來創建字典：

In [2]:
df = pd.DataFrame(columns=list('abcdefg'))
df

Out[2]:
Empty DataFrame
Columns: [a, b, c, d, e, f, g]
Index: []

In [3]:
new_cols = dict(zip(df.columns[2:5], ['KOAK-' + x for x in df.columns[2:5]]))
new_cols

Out[3]:
{'c': 'KOAK-c', 'd': 'KOAK-d', 'e': 'KOAK-e'}

In [6]:    
df.rename(columns= new_cols, inplace=True)
df

Out[6]:
Empty DataFrame
Columns: [a, b, KOAK-c, KOAK-d, KOAK-e, f, g]
Index: []

因此，在您的護理下，以下方法應該起作用：

new_cols = dict(zip(df.columns[3:47], ['KOAK-' + str(x) for x in df.columns[3:47]]))
df.rename(columns= new_cols, inplace=True)