python实现比较两个csv文件中的指定的两列数据,将相等的数据打印出来
时间: 2023-05-24 19:07:01 浏览: 185
比较任意两个csv文件
3星 · 编辑精心推荐
可以使用pandas库来处理csv文件,具体实现步骤如下:
1. 读入两个csv文件,存储为两个DataFrame对象
```python
import pandas as pd
df1 = pd.read_csv('file1.csv')
df2 = pd.read_csv('file2.csv')
```
2. 指定需要比较的两列数据,使用merge函数将两个DataFrame对象按照这两列进行合并
```python
col1 = 'col1'
col2 = 'col2'
merged = pd.merge(df1, df2, on=[col1, col2], how='inner')
```
`on`参数指定按照哪两列进行合并,`how`参数指定使用内连接,即只保留相等的数据。
3. 打印合并后的数据
```python
print(merged)
```
如果需要将合并后的数据存储为新的csv文件,可以使用to_csv函数。
```python
merged.to_csv('merged.csv', index=False)
```
阅读全文