python根据csv数据名称和数据大小筛选保存到新csv里代码
时间: 2023-09-17 15:12:52 浏览: 79
可以使用Python中的pandas库来读取和处理csv文件,然后根据数据名称和大小进行筛选并保存到新的csv文件中。以下是一个示例代码:
```python
import pandas as pd
import os
# 设置需要筛选的数据名称和大小
data_name = 'example.csv'
data_size = 1000
# 设置读取和保存文件的路径
data_path = 'path/to/csv/files'
new_data_path = 'path/to/new/csv/file'
# 获取文件列表
file_list = os.listdir(data_path)
# 遍历文件列表
for file_name in file_list:
# 仅处理csv文件
if file_name.endswith('.csv'):
# 读取csv文件
file_path = os.path.join(data_path, file_name)
df = pd.read_csv(file_path)
# 筛选数据
df = df[(df['name'] == data_name) & (df['size'] == data_size)]
# 保存新的csv文件
if not df.empty:
new_file_name = 'filtered_' + file_name
new_file_path = os.path.join(new_data_path, new_file_name)
df.to_csv(new_file_path, index=False)
```
该代码遍历指定路径下的所有csv文件,读取每个csv文件并筛选出名称为`data_name`且大小为`data_size`的数据,然后将筛选结果保存到新的csv文件中。请根据实际情况修改代码中的路径和筛选条件。