python 向csv文件中插入一列新数据

产生一组随机数据，然后将之插入进csv文件中，插入的位置为最后一列。
代码如下：

import csv
import random

with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    data = list(csv_reader)

# 添加行数据
for row in data:
    modify = random.randint(1,3)
    row.append(modify)

# 写入新的CSV文件
with open('data.csv', 'w', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(data)