pandas使用字典格式修改columns列名

核心方法是:

s = df.columns.to_series()
df.columns = s.map(dic).fillna(s)

使用示例

import pandas as pd

if __name__ == '__main__':
    df = pd.DataFrame([
        [0, 1, 1],
        [0, 1, 1],
        [0, 2, 2],
        [0, 2, 3]
    ], columns=['a', 'b', 'c'])
    dic = {
    
    'a': '1', "b": "2"}
    s = df.columns.to_series()
    df.columns = s.map(dic).fillna(s)
    print(df.columns.tolist())

结果:['1', '2', 'c']

猜你喜欢

转载自blog.csdn.net/weixin_35757704/article/details/121639478