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

posted on 2020-07-28 17:43  疯狂的小萝卜头  阅读(6794)  评论(0编辑  收藏  举报