统计csv文件A的列数据,在csv文件B的指定列出现的次数,并写入到新的文件;
如A文件数据:
1,智能,pjfw,PJFW,56001,PJFW,pjfw,123456,SYCZ,,- 2,智能,文,SDZP,56003,SDZZ,test01,123456,ZBHT,订阅,
如B文件数据:
1,PJZN,蓝 2,SYCZ,蓝
import csv
file_path1 = "site_ori.csv"
file_path2 = "site_origin.csv"
file_path3 = "site.csv"
with open(file_path1, "r", encoding='utf-8') as fp1, \
open(file_path2, "r", encoding='utf-8') as fp2, \
open(file_path3, "a+", encoding='utf-8', newline='') as fp3:
# 读取文件
readcsv1 = csv.reader(fp1)
readcsv2 = csv.reader(fp2)
writercsv3 = csv.writer(fp3)
# 第2个文件的第2列的数据,与第1个文件的第9列的数据比较,统计出现的次数
for row2 in readcsv2:
count = 0
for row1 in readcsv1:
# print(row1[8], row2[1])
if (row1[8] == row2[1]):
count += 1
# print(count)
writercsv3.writerow([row2[0], row2[1], row2[2], count])
# 将文件1的指针移动到起始位置(第一行)
fp1.seek(0)