请写一段代码，使用python读取train.csv和output.csv并将train.csv的前三列与output.csv的第一列进行拼接，输出一个新的csv文件

以下是使用Python读取train.csv和output.csv，将train.csv的前三列与output.csv的第一列进行拼接，输出一个新的csv文件的代码：

import csv

# 读取train.csv文件
with open('train.csv', 'r') as train_file:
    train_reader = csv.reader(train_file)
    train_data = [row[:3] for row in train_reader]  # 取前三列数据

# 读取output.csv文件
with open('output.csv', 'r') as output_file:
    output_reader = csv.reader(output_file)
    output_data = [row[:1] for row in output_reader]  # 取第一列数据

# 拼接数据
new_data = []
for i in range(len(train_data)):
    new_row = train_data[i] + output_data[i]
    new_data.append(new_row)

# 写入新的csv文件
with open('new_data.csv', 'w', newline='') as new_file:
    writer = csv.writer(new_file)
    for row in new_data:
        writer.writerow(row)

其中，`csv.reader()`方法用于读取csv文件，`csv.writer()`方法用于写入csv文件。`newline=''`参数用于避免写入csv文件时出现空行。最后，使用`writerow()`方法将每一行写入新的csv文件中。