如何使用python在csv文件中重复行n次

import csv

with open('example.csv', 'r') as file:
    reader = csv.reader(file)
    rows = list(reader)

new_rows = []

for row in rows:

    repeated_row = [row] * n
    new_rows.extend(repeated_row)

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(new_rows)

import csv

def repeat_rows(input_file, output_file, n):
    with open(input_file, 'r') as file:
        reader = csv.reader(file)
        rows = list(reader)
    
    new_rows = []
    
    for row in rows:
        repeated_row = [row] * n
        new_rows.extend(repeated_row)
    
    with open(output_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(new_rows)

# 调用函数进行重复行操作
repeat_rows('example.csv', 'output.csv', 3)