python提取csv文件某几行代码
时间: 2023-12-03 19:04:30 浏览: 84
源代码.txt
假设你已经使用 `csv` 模块将 CSV 文件读入到了程序中并存储为一个名为 `csv_reader` 的变量。接下来,你可以使用以下代码提取 CSV 文件中的某几行:
```python
# 从 csv_reader 中提取前两行数据
first_two_rows = []
for i, row in enumerate(csv_reader):
if i < 2:
first_two_rows.append(row)
else:
break
```
上述代码中,我们使用了 `enumerate` 函数来获取每一行的索引 `i` 和对应的行数据 `row`。我们只需要在循环中判断当前的索引是否小于要提取的行数,如果小于,则将当前行添加到一个列表中,否则退出循环。在本例中,我们提取了前两行数据,并将其存储到了 `first_two_rows` 列表中。你可以根据需要修改代码来提取 CSV 文件中的其他行。
阅读全文