Python查询csv符合条件的行

在 Python 中，可以使用标准库的 csv 模块来读取 CSV 文件，并查询符合条件的行。下面是一个示例代码，展示如何查询 CSV 文件中符合条件的行：

假设有一个名为 data.csv 的 CSV 文件，它的内容类似于这样：

Name, Age, City
John, 25, New York
Alice, 30, Los Angeles
Bob, 22, Chicago
Eve, 28, New York

现在希望查询所有居住在 "New York" 的人的信息。

import csv

def query_csv(file_path, city):
    matching_rows = []

    with open(file_path, newline='') as csvfile:
        csv_reader = csv.DictReader(csvfile)
        for row in csv_reader:
            if row['City'] == city:
                matching_rows.append(row)

    return matching_rows

file_path = 'data.csv'
city_to_query = 'New York'
result = query_csv(file_path, city_to_query)

for row in result:
    print(row['Name'], row['Age'], row['City'])

现在希望查询所有居住在 "New York" 的人的信息。

import csv

def query_csv(file_path, city):
    matching_rows = []

    with open(file_path, newline='') as csvfile:
        csv_reader = csv.DictReader(csvfile)
        for row in csv_reader:
            if row['City'] == city:
                matching_rows.append(row)

    return matching_rows

file_path = 'data.csv'
city_to_query = 'New York'
result = query_csv(file_path, city_to_query)

for row in result:
    print(row['Name'], row['Age'], row['City'])

运行这个代码，输出将会是：

John 25 New York
Eve 28 New York

这个例子中，使用了 csv.DictReader 来读取 CSV 文件，它会将每一行数据转换为一个字典，其中字典的键是 CSV 文件的第一行的标题（Name、Age、City），而字典的值是对应行的数据。然后遍历每一行，检查是否满足查询条件（居住城市为 "New York"），如果满足，则将该行添加到 matching_rows 列表中。最后，输出符合条件的行的信息。

这样，就可以根据需要查询 CSV 文件中符合条件的行。