如何根据条件替换csv文件中的数据？

import csv

def replace_csv_data(csv_file, condition, replacement):
    # 读取CSV文件
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        data = list(reader)

    # 根据条件进行数据替换
    for row in data:
        if condition(row):
            # 根据条件替换数据
            replacement(row)

    # 更新CSV文件
    with open(csv_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerows(data)

# 示例条件：替换第一列为"apple"的行的第二列数据为"red"
def condition(row):
    return row[0] == "apple"

def replacement(row):
    row[1] = "red"

# 替换CSV文件中的数据
replace_csv_data('data.csv', condition, replacement)