我如何重构这段特定的代码？

Question

我是编程新手，目前正在重构这段代码，这段代码的作用是将列名中的空格替换为'_'，代码如下：

labels = list(df.columns)
labels[0] = labels[0].replace(' ', '_')
labels[1] = labels[1].replace(' ', '_')
labels[2] = labels[2].replace(' ', '_')
labels[3] = labels[3].replace(' ', '_')
labels[5] = labels[5].replace(' ', '_')
labels[6] = labels[6].replace(' ', '_')
df.columns = labels

我如何重构它并使其可重用？提前致谢。

Answer 1

试试这个：

def fun(label):
    return label.replace(' ', '_')
df.columns = map(fun, df.columns)