在列中找到重复值的情况下导出csv行

import csv

def find_duplicate_rows(input_file, output_file, column_index):
    seen = set()
    duplicates = []

    with open(input_file, mode='r', newline='', encoding='utf-8') as infile:
        reader = csv.reader(infile)
        header = next(reader)  # 读取表头

        for row in reader:
            value = row[column_index]
            if value in seen:
                duplicates.append(row)
            else:
                seen.add(value)

    with open(output_file, mode='w', newline='', encoding='utf-8') as outfile:
        writer = csv.writer(outfile)
        writer.writerow(header)  # 写入表头
        writer.writerows(duplicates)

# 使用示例
input_csv = 'input.csv'
output_csv = 'duplicates.csv'
column_to_check = 2  # 假设我们要检查第三列（索引为2）是否有重复值
find_duplicate_rows(input_csv, output_csv, column_to_check)