首页 > 编程知识 正文

Python合并两个csv文件

时间:2023-11-22 00:39:47 阅读:298537 作者:BWCG

在本篇文章中,我们将探讨如何使用Python合并两个csv文件。

一、读取CSV文件

在开始合并两个csv文件之前,我们需要先读取这两个文件的内容。Python提供了csv模块来处理csv文件,我们可以使用该模块中的函数来读取csv文件的内容。

import csv

# 读取第一个csv文件
csv_file1 = open('file1.csv', 'r')
reader1 = csv.reader(csv_file1)
data1 = list(reader1)

# 读取第二个csv文件
csv_file2 = open('file2.csv', 'r')
reader2 = csv.reader(csv_file2)
data2 = list(reader2)

csv_file1.close()
csv_file2.close()

在上面的代码中,我们使用了csv模块的reader函数来读取csv文件,并将文件内容存储在一个二维列表中。需要注意的是,我们在读取完csv文件后要记得关闭文件。

二、合并CSV文件

在读取完两个csv文件后,我们可以对它们进行合并。合并的方式有很多种,这里我们选择将两个文件的内容按行合并。

merged_data = data1 + data2

在上面的代码中,我们使用了“+”运算符将两个二维列表合并成一个。合并后的结果存储在merged_data变量中。

三、写入CSV文件

合并完csv文件的内容后,我们可以将结果写入一个新的csv文件中。

csv_file_merged = open('merged_file.csv', 'w', newline='')
writer = csv.writer(csv_file_merged)
writer.writerows(merged_data)

csv_file_merged.close()

在上面的代码中,我们使用了csv模块的writer函数来写入csv文件。需要注意的是,我们在写入csv文件之前要先打开文件,并且在写入完毕后要记得关闭文件。

四、完整代码示例

import csv

# 读取第一个csv文件
csv_file1 = open('file1.csv', 'r')
reader1 = csv.reader(csv_file1)
data1 = list(reader1)

# 读取第二个csv文件
csv_file2 = open('file2.csv', 'r')
reader2 = csv.reader(csv_file2)
data2 = list(reader2)

csv_file1.close()
csv_file2.close()

# 合并CSV文件
merged_data = data1 + data2

# 写入CSV文件
csv_file_merged = open('merged_file.csv', 'w', newline='')
writer = csv.writer(csv_file_merged)
writer.writerows(merged_data)

csv_file_merged.close()

通过以上代码,我们可以轻松地使用Python合并两个csv文件。

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。