📜  csv.DictReader 跳过行 - Python (1)

📅  最后修改于: 2023-12-03 15:14:23.702000             🧑  作者: Mango

Python中使用csv.DictReader跳过行

在处理大量数据时,使用Python的csv模块进行读写是非常常见的。其中,csv.DictReader是一个很实用的工具,能够将CSV文件逐行读入为字典,方便我们处理CSV数据。然而,在某些情况下,我们需要跳过CSV文件的某些行,本文将介绍如何使用csv.DictReader跳过行。

1. 跳过首行

在处理CSV文件时,通常首行是描述数据的标题。如果不希望在字典中包含标题,可以使用csv.DictReader的next()方法跳过首行。具体实现如下:

import csv

with open('data.csv', 'r') as f:
    csv_reader = csv.DictReader(f)
    # 跳过首行
    next(csv_reader)
    for row in csv_reader:
        print(row)
2. 跳过指定行

有时,我们需要跳过文件中的指定行。可以使用enumerate()函数来获得当前读入的行数,然后通过判断来跳过指定行。具体实现如下:

import csv

with open('data.csv', 'r') as f:
    csv_reader = csv.DictReader(f)
    # 跳过第3行和第4行
    for i, row in enumerate(csv_reader):
        if i == 2 or i == 3:
            continue
        print(row)
3. 结语

csv.DictReader是一个很实用的工具,方便我们将CSV文件逐行读入为字典。使用以上方法可以让读入CSV文件更加灵活,满足更多的需求。