向多个csv文件添加标头

import csv
import os

def add_header_to_csv(file_path, header):
    # 打开原始csv文件
    with open(file_path, 'r') as csvfile:
        reader = csv.reader(csvfile)
        data = [row for row in reader]
    
    # 创建新的csv文件并写入标头
    new_file_path = os.path.splitext(file_path)[0] + '_new.csv'
    with open(new_file_path, 'w', newline='') as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(header)
        
        # 写入原始csv文件的内容
        writer.writerows(data)

# 调用函数添加标头
file_paths = ['file1.csv', 'file2.csv', 'file3.csv']
header = ['Column1', 'Column2', 'Column3']
for file_path in file_paths:
    add_header_to_csv(file_path, header)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;

public class AddHeaderToCSV {
    public static void main(String[] args) {
        String[] filePaths = {"file1.csv", "file2.csv", "file3.csv"};
        String[] header = {"Column1", "Column2", "Column3"};
        
        for (String filePath : filePaths) {
            addHeaderToCSV(filePath, header);
        }
    }

    public static void addHeaderToCSV(String filePath, String[] header) {
        try (BufferedReader reader = new BufferedReader(new FileReader(filePath));
             BufferedWriter writer = new BufferedWriter(new FileWriter(filePath + "_new.csv"))) {
            
            // 写入标头
            for (int i = 0; i < header.length; i++) {
                writer.write(header[i]);
                if (i < header.length - 1) {
                    writer.write(",");
                }
            }
            writer.newLine();
            
            // 写入原始csv文件的内容
            String line;
            while ((line = reader.readLine()) != null) {
                writer.write(line);
                writer.newLine();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}