整合多个CSV文件

import cv2
import os
import csv
import pandas as pd
import random
import shutil

#先整合需要抽样的CSV文件
#需要整合的CSV文件存放在该路径下
data_path = '../record2fen/'
img_list = os.listdir(data_path)

#需要保存的CSV文件存放在该路径下
save_path = '../record1000/'
if not os.path.exists(save_path):
    os.mkdir(save_path)
#row存放合并后的CSV文件
row = []
for i in range(0, len(img_list)):
    img_full_path = os.path.join(data_path, img_list[i])
    #and后面的可以修改选择自己想要的进行合并
    if os.path.exists(img_full_path) and img_full_path.endswith('.csv'): # 关键筛选
        record_csv = pd.read_csv(img_full_path)
        for j in range(len(record_csv)):
            row.append(record_csv.iloc[j])
#print(len(row))

with open(os.path.join(save_path, "record.csv"),'a+',newline='') as t:
    #创建初始化写入对象
    writer=csv.writer(t)
     #创建初始化写入对象S
    writer.writerows(row)