在Python中,可以使用csv模块和pandas库来将csv文件中的两列数据添加到同一csv文件的新列中。
使用csv模块的示例代码如下:
import csv
input_file = 'input.csv'
output_file = 'output.csv'
col1 = 'Column1'
col2 = 'Column2'
new_col = 'NewColumn'
# 打开输入和输出文件
with open(input_file, 'r') as file_in, open(output_file, 'w', newline='') as file_out:
# 创建CSV读取器和写入器
reader = csv.DictReader(file_in)
fieldnames = reader.fieldnames + [new_col] # 添加新列名到fieldnames
writer = csv.DictWriter(file_out, fieldnames=fieldnames)
# 写入表头
writer.writeheader()
# 逐行处理数据
for row in reader:
# 将两列数据拼接为新列数据
new_data = row[col1] + row[col2]
row[new_col] = new_data
# 写入新行
writer.writerow(row)
使用pandas库的示例代码如下:
import pandas as pd
input_file = 'input.csv'
output_file = 'output.csv'
col1 = 'Column1'
col2 = 'Column2'
new_col = 'NewColumn'
# 读取CSV文件并添加新列
df = pd.read_csv(input_file)
df[new_col] = df[col1] + df[col2]
# 保存为新的CSV文件
df.to_csv(output_file, index=False)
以上代码示例中,input.csv
是输入的CSV文件,Column1
和Column2
是需要合并的两列数据,NewColumn
是新的列名,output.csv
是输出的CSV文件。
推荐的腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云