读取两个csv文件,选择两个文件中列名相同的两列,逐行匹配两列数据相同的行写入新文件
时间: 2024-05-12 19:18:28 浏览: 183
Java中easypoi导入excel文件列名相同的处理方案
好的,这是一个很常见的数据处理问题,你可以使用Python语言进行处理。可以使用pandas库来读取csv文件,如果需要安装,可以使用pip install pandas命令安装。下面是个例子:
```python
import pandas as pd
# 读取csv文件
file1 = pd.read_csv('file1.csv')
file2 = pd.read_csv('file2.csv')
# 获取列名相同的列
common_columns = set(file1.columns).intersection(set(file2.columns))
# 定义一个空的DataFrame
result = pd.DataFrame()
# 遍历列名相同的列
for column in common_columns:
# 获取两个文件中的同名列
column1 = file1[column]
column2 = file2[column]
# 找到相同的行并添加到结果DataFrame
matches = file1[column1.isin(column2)]
result = pd.concat([result, matches])
# 保存结果到新文件
result.to_csv('result.csv', index=False)
```
以上代码假设你的csv文件中没有标题行,如果有,可以去掉代码中的header参数或者将其设置为0。请注意,这只是一个简单的例子,如果两个文件非常大,这种方法可能会很慢,你可能需要考虑使用更高效的方法来处理数据。
阅读全文