用python筛选csv特定行
时间: 2023-06-14 15:05:37 浏览: 182
可以使用Python内置的csv模块来读取和筛选csv文件中的特定行。
以下是一个示例代码,它将读取名为“data.csv”的csv文件,并且只保留特定条件下的行,然后将结果保存到名为“filtered_data.csv”的新文件中:
```python
import csv
# 打开csv文件
with open('data.csv', 'r') as csv_file:
# 读取csv文件
csv_reader = csv.reader(csv_file)
# 创建一个新的csv文件,用于存储筛选后的结果
with open('filtered_data.csv', 'w', newline='') as new_csv_file:
csv_writer = csv.writer(new_csv_file)
# 遍历每一行数据
for row in csv_reader:
# 如果行满足特定条件,则将其写入新文件
if row[0] == '特定条件':
csv_writer.writerow(row)
```
上述代码中,我们使用`csv.reader`函数来读取csv文件中的每一行数据,并使用`csv.writer`函数将符合特定条件的行写入到新文件中。
需要注意的是,`csv.reader`和`csv.writer`返回的都是可迭代的对象,因此可以通过`for`循环遍历每一行数据。同时,在打开文件时,我们使用了`with`语句来确保在使用完文件后自动关闭它们。
阅读全文