如何在Python(Pandas)中重命名特定范围的列

时间:2016-01-16 20:32:57

标签: python python-3.x pandas

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

以上代码允许我在数据框中的所有列标题前添加“KOAK”,但我只想将KOAK前缀添加到列号4-38。我该怎么做呢?

2 个答案:

答案 0 :(得分:0)

您可以通过使用列表推导来压缩列序号范围并将其传递给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)

答案 1 :(得分:0)

您可以分配到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