在CSV文件之间复制特定行Python

import csv

def copy_specific_rows(source_file, destination_file, rows_to_copy):
    with open(source_file, 'r') as source, open(destination_file, 'w', newline='') as destination:
        reader = csv.reader(source)
        writer = csv.writer(destination)
        
        for i, row in enumerate(reader):
            if i in rows_to_copy:
                writer.writerow(row)

# 示例用法
source_file = 'source.csv'
destination_file = 'destination.csv'
rows_to_copy = [1, 3, 5]  # 要复制的行的索引

copy_specific_rows(source_file, destination_file, rows_to_copy)

import pandas as pd

def copy_specific_rows(source_file, destination_file, rows_to_copy):
    df = pd.read_csv(source_file)
    df_selected = df.iloc[rows_to_copy]
    df_selected.to_csv(destination_file, index=False)

# 示例用法
source_file = 'source.csv'
destination_file = 'destination.csv'
rows_to_copy = [1, 3, 5]  # 要复制的行的索引

copy_specific_rows(source_file, destination_file, rows_to_copy)