python批量读取csv并提取其中多行生成新csv
Posted lwx2233
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了python批量读取csv并提取其中多行生成新csv相关的知识,希望对你有一定的参考价值。
同学有大量csv文件,要我帮忙提取每个csv中‘type’字段为‘O3’的整一行数据
于是,上代码!
import csv
import os
import pandas as pd
import json
# 设置显示的最大列、宽等参数,消掉打印不完全中间的省略号
# pd.set_option('display.max_columns', 1000)
pd.set_option('display.width', 1000)
# pd.set_option('display.max_colwidth', 1000)
# pd.set_option('display.height', 1000)
# 显示所有列
pd.set_option('display.max_columns', None)
# 显示所有行
pd.set_option('display.max_rows', None)
dict = []
file_path = 'file' # 设置文件路径
# 建立空list
code_list = []
for root, dirs, files in os.walk(file_path):
# print(root)
# print(dirs)
# print(files)
for filename in files:
code_list.append(filename)
list_number = len(code_list)
index = 0
for k in code_list:
if index < int(list_number):
file_name = 'file/' + code_list[index]
index += 1
with open(file_name, 'r', encoding="utf-8") as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
if row['type'] == 'O3':
# j = json.dumps(row, ensure_ascii=False)
# print(j)
# with open('test_data.json', 'a') as json_file:
# json_file.write(j)
dict.append(row)
print(row)
pd.DataFrame(dict).to_csv('2021.csv')
利用pandas的DataFrame.to_csv方法,记得用的输入参数是字典格式而并非json
以上是关于python批量读取csv并提取其中多行生成新csv的主要内容,如果未能解决你的问题,请参考以下文章