python实现比较两个csv文件中的两列数据,并将其相等的数据打印出来
时间: 2023-05-24 14:06:53 浏览: 293
利用Python如何实现数据驱动的接口自动化测试
实现代码如下:
```python
import csv
# 打开两个csv文件
file1 = open('file1.csv', 'r')
file2 = open('file2.csv', 'r')
# 读取csv文件中需要比较的列
col1 = csv.reader(file1)
col2 = csv.reader(file2)
# 新建一个字典存储两列数据相等的内容
same_content = {}
# 遍历两列数据并比较其内容
for row1 in col1:
for row2 in col2:
if row1[1] == row2[1]: # 比较第2列数据
same_content[row1[1]] = [row1[0], row2[0]] # 将相等的数据存入字典中
break
# 关闭csv文件
file1.close()
file2.close()
# 打印出两列数据相等的内容
for key, value in same_content.items():
print(key, value[0], value[1])
```
上述代码中,我们首先打开需要比较的两个csv文件,读取两列数据,然后使用两个循环遍历两列数据,比较第二列数据是否相等,相等则将其存入字典中。最后,遍历字典并打印出相等的两列数据。
阅读全文