KOAK.rename(columns = lambda x : 'KOAK-' + x, inplace=True)
以上代码允许我在数据框中的所有列标题前添加“KOAK”,但我只想将KOAK前缀添加到列号4-38。我该怎么做呢?
答案 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