csv文件的读写-pandas和python csv包两个版本

csv文件的读写-pandas和python csv包两个版本

python有csv可以直接import用于读写csv文件，pandas包也可以直接读取csv文件。

import csv 读写csv文件

主要使用reader和writer读写文件

from tqdm import tqdm
filename = "./test.csv"
csvFile = open(filename, "r")
reader = csv.reader(csvFile)

fileHeader = []
csvWriter = open("output.csv", 'w')
writer = csv.writer(csvWriter)

# 将输入文件对应行的处理结果写到输出文件中
for item in tqdm(reader):
    if reader.line_num == 1:
        fileHeader = ['c1','c2','c3'] # 需要写的csv文件的列名
        print("fileHeader:", len(fileHeader), fileHeader)
        writer.writerow(fileHeader)
	a = item[0]
	b = item[1]
	c = item[2]
	writer.writerow([a,b,c])
	

pandas读写csv文件

import pandas as pd
table = pd.Dataframe(columns=["c1", "c2", 'c3']) # 自定义写文件
# df = pd.read_csv("./test.csv") # 读文件
# 读文件报错可尝试：df = pd.read_csv("./test.csv",lineterminator='n')

table = table.append({'c1': a, 'c2': b, 'c3': b}, ignore_index=True)
table.to_csv("output.csv")