下面是一种比较两个CSV文件中的两列数据的解决方法,包含代码示例:
import csv
def compare_columns(file1, file2, column1, column2):
with open(file1, 'r') as csv_file1, open(file2, 'r') as csv_file2:
reader1 = csv.DictReader(csv_file1)
reader2 = csv.DictReader(csv_file2)
data1 = [row[column1] for row in reader1]
data2 = [row[column2] for row in reader2]
# 比较两列数据是否相等
if data1 == data2:
print("两列数据相等")
else:
print("两列数据不相等")
# 示例用法
compare_columns('file1.csv', 'file2.csv', 'column1', 'column2')
在这个示例中,我们使用csv
模块读取两个CSV文件,并获取指定列的数据。然后,我们比较这两列数据是否相等。如果相等,输出"两列数据相等",否则输出"两列数据不相等"。
请确保将file1.csv
和file2.csv
替换为你自己的CSV文件路径,column1
和column2
替换为你想要比较的列的名称。