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