python 利用pandas读取本地中CSV文件的指定列 列名重命名 并保存回本地

需求

读取本地中CSV文件的指定列，并对列进行重命名，并保存回本地

原数据展示

movies.csv

操作后数据展示

new_movies.csv

代码

# -*- coding: utf-8 -*-
import json
import pandas as pd

# 所需列名和新老列名映射关系
columns_json_str = '{"name":"NEW_NAME","src":"NEW_SRC"}'
columns_dict = json.loads(columns_json_str)

# 读取本地文件
dataset = pd.read_csv('movies.csv', header=0, encoding='utf-8', dtype=str)
df = pd.DataFrame(dataset, columns=columns_dict.keys())

# 列名重命名
# df.rename(columns={'name':'NEW_NAME', 'src':'NEW_SRC'}, inplace=True) 
df.rename(columns=columns_dict, inplace=True) 
print(df)

# 将重命名之后的数据写入到文件
filepath = 'new_movies.csv'
df_columns = pd.DataFrame([list(df.columns)])
df_columns.to_csv(filepath, mode='w', header=False, index=0) 

df.to_csv(filepath, mode='a', header=False, index=0)

Git地址

https://github.com/gm19900510/data_analysis_python 欢迎star