python筛选符合条件的行代码
时间: 2023-09-24 21:04:02 浏览: 72
以下是示例代码,用于在Python中筛选符合条件的行:
```python
import csv
def filter_rows(csv_file, condition):
filtered_rows = []
with open(csv_file, 'r') as file:
reader = csv.reader(file)
header = next(reader) # 获取表头
filtered_rows.append(header) # 添加表头到筛选结果
for row in reader:
if condition(row): # 判断行是否满足条件
filtered_rows.append(row) # 添加满足条件的行到筛选结果
return filtered_rows
# 示例条件函数:筛选年龄大于等于18岁的行
def age_condition(row):
age = int(row[2]) # 假设年龄在第3列(索引为2)
return age >= 18
# 使用示例
csv_file = 'data.csv' # 替换为你的CSV文件路径
filtered_rows = filter_rows(csv_file, age_condition)
for row in filtered_rows:
print(row)
```
你可以根据自己的需求修改条件函数 `age_condition`,以及CSV文件路径 `csv_file`。注意,这里假设CSV文件的第一行是表头,数据从第二行开始。
阅读全文