Python CSV 读取特定行

在本文中，我们将介绍如何使用 Python 读取 CSV 文件中的特定行。CSV（逗号分隔值）是一种常见的文件格式，用于存储表格数据。Python 提供了强大的内置库 csv，可以轻松读取和处理 CSV 文件。

CSV 文件介绍

CSV 文件是以纯文本形式存储的表格数据。每行代表一行数据，每个字段之间用逗号分隔。CSV 文件可以使用电子表格软件（如 Microsoft Excel、Google Sheets）或文本编辑器创建和编辑。以下是一个示例 CSV 文件的内容：

Name,Age,City
Alice,25,New York
Bob,30,San Francisco
Charlie,35,Los Angeles

在这个示例中，每行都包含三个字段：姓名、年龄和城市。

读取特定行

要读取特定行，我们可以使用 Python 的 csv 模块。首先，我们需要打开 CSV 文件并创建一个 csv.reader 对象：

import csv

with open('data.csv', 'r') as file:
    reader = csv.reader(file)

在打开文件时，我们使用了 'r' 参数来指定只读模式。然后，我们创建了一个 csv.reader 对象来读取文件的内容。接下来，我们可以使用 next() 函数跳过指定数量的行。例如，如果我们想要跳过第一行的标题，可以这样写：

next(reader)  # 跳过第一行

现在，我们已经跳过了第一行，可以继续读取特定行。这里有两种常见的方法来读取特定行：按行号读取和按条件读取。

按行号读取特定行

我们可以使用 islice() 函数从 csv.reader 对象中按行号读取特定行。以下是一个例子，读取第三行（索引从零开始）：

from itertools import islice

row_number = 2  # 第三行（索引为2）
specific_row = list(islice(reader, row_number, row_number+1))

在这个例子中，我们使用 islice() 函数并将 reader 对象作为第一个参数。然后，我们指定要读取的行号范围，这里是从第三行到第四行（结束位置不包含在内）。specific_row 变量将包含特定行的数据。

按条件读取特定行

除了按行号读取特定行外，我们还可以按照某个条件来读取特定行。例如，假设我们只想读取居住在纽约的人的数据。我们可以使用 csv.reader 对象的 if 语句来筛选符合条件的行。以下是一个示例：

specific_rows = [row for row in reader if row[2] == 'New York']

在这个示例中，我们使用了一个列表推导式来循环遍历 reader 对象的每一行。如果某一行的第三个字段（城市）等于 'New York'，则将该行添加到 specific_rows 列表中。

完整示例

下面是一个完整的示例，演示了如何读取 CSV 文件的特定行：

import csv
from itertools import islice

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    next(reader)  # 跳过第一行

    row_number = 2  # 第三行（索引为2）
    specific_row = list(islice(reader, row_number, row_number+1))
    print(specific_row)

    file.seek(0)  # 将文件指针重置到开头
    next(reader)  # 跳过第一行

    specific_rows = [row for row in reader if row[2] == 'New York']
    print(specific_rows)

这个示例首先打开名为 data.csv 的 CSV 文件，并跳过第一行（标题）。然后，它使用 islice() 函数读取第三行，并将结果存储在 specific_row 变量中。接下来，它将文件指针重置到开头，并再次跳过第一行。最后，它使用列表推导式找出居住在纽约的人的数据，并将结果存储在 specific_rows 列表中。