python将一个csv文件中的一列数据作为条件,利用这些条件去筛选另一个csv文件中对应的数据,并将这些数据按照条件对应的数据各自存成新的csv文件
时间: 2023-05-25 13:05:52 浏览: 306
pandas读取CSV文件时查看修改各列的数据类型格式
5星 · 资源好评率100%
可以使用pandas库来完成这个任务。下面是一个简单的代码示例:
```python
import pandas as pd
# 读取包含条件的csv文件
condition_df = pd.read_csv('condition.csv')
# 读取需要筛选的csv文件
data_df = pd.read_csv('data.csv')
# 将条件列设置为索引
condition_df.set_index('条件列名', inplace=True)
# 根据条件筛选数据
filtered_data = data_df[data_df['要筛选的列名'].isin(condition_df.index)]
# 按照条件对应的数据存储为新的csv文件
for condition_value, group_df in filtered_data.groupby('要筛选的列名'):
group_df.to_csv(f'{condition_value}.csv', index=False)
```
其中,`condition.csv`包含一列条件值,`data.csv`里包含需要筛选的数据。`filtered_data`是根据条件筛选出来的数据。最后,`filtered_data`根据条件对应的数据,将数据分组并存储为新的csv文件。
阅读全文