python提取csv文件不连续的几行代码
时间: 2024-03-02 16:22:16 浏览: 65
py代码-处理csv文件重复数据
可以使用Python的csv模块来读取csv文件中的数据,然后使用切片操作来提取不连续的几行数据。
下面是一个示例代码,假设csv文件中有10行数据,要提取第1、3和5行数据:
```python
import csv
with open('data.csv', 'r') as file:
reader = csv.reader(file)
rows = [row for idx, row in enumerate(reader) if idx in (0, 2, 4)]
print(rows)
```
这里使用了一个列表推导式来遍历csv文件中的每一行数据,然后使用`enumerate()`函数获取每行数据的索引,判断是否为需要提取的行号,如果是则将该行数据添加到一个列表中。
需要注意的是,csv文件中的数据可能包含有换行符,需要使用`strip()`函数来去除每行数据中的多余空格和换行符。同时,如果csv文件中包含有标题行,需要在提取数据时将其排除,比如上面示例代码中的`(0, 2, 4)`就是排除了标题行后需要提取的行号。
阅读全文