将字典中的值与数据帧行值进行匹配，并将数据添加到该行

import csv

data_dict = {
    'key1': 'value1',
    'key2': 'value2',
    'key3': 'value3'
}

# 读取数据文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    data = list(reader)

# 处理每一行数据
for row in data:
    # 获取行值
    row_value = row[0]
    
    # 检查行值是否存在于字典的键中
    if row_value in data_dict:
        # 将字典中对应的值添加到该行的末尾
        row.append(data_dict[row_value])

# 将更新后的数据写回到原始数据文件
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)