Python:替换csv的某些行，但不替换其他行

import csv

with open('input.csv', 'r') as input_file, open('output.csv', 'w', newline='') as output_file:
    reader = csv.reader(input_file)
    writer = csv.writer(output_file)

    for row in reader:
        # 判断是否需要替换该行
        if condition:
            # 替换该行的内容
            row = new_row
        # 写入目标CSV文件
        writer.writerow(row)

input_file.close()
output_file.close()

import csv

with open('input.csv', 'r') as input_file, open('output.csv', 'w', newline='') as output_file:
    reader = csv.reader(input_file)
    writer = csv.writer(output_file)

    for row in reader:
        # 判断是否需要替换该行
        if condition:
            # 替换该行的内容
            row = new_row
        # 写入目标CSV文件
        writer.writerow(row)

input_file.close()
output_file.close()