python CSV 文件读取 存储 例子

我想读取一个CSV文件，然后只取前几条数据，之后存储成为一个新的文件，代码很简单，注意看注释哦

import csv

#这个是我要读取的csv，文件名不一定携带csv也可以
file_dir = "./666.c000"

#这个是我要写入的
write_name = "./csvSamples.csv"
csvFile = open(file_dir,"r")
csvFileS = open(write_name,"w")
#读取
reader = csv.DictReader(csvFile)
field_name = reader.fieldnames
#建立一个写入的笔
writer = csv.DictWriter(csvFileS,field_name)
threshold = 300
#我筛选了300行
i = 1
for row in reader:
    writer.writerow(row)
    i+=1
    if i >threshold:
        break

参考：https://blog.csdn.net/guoziqing506/article/details/52014506