Как я могу реорганизовать этот конкретный фрагмент кода?

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)