python从csv提取特定数据

在Python中，可以使用标准库的csv模块来从CSV文件中提取特定数据。假设有一个名为data.csv的CSV文件，包含以下内容：

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

现在，希望从CSV文件中提取居住在 "New York" 的人的姓名和年龄信息。

import csv

def extract_data_by_city(file_path, city):
    extracted_data = []

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

    return extracted_data

file_path = 'data.csv'
city_to_extract = 'New York'
result = extract_data_by_city(file_path, city_to_extract)

for name, age in result:
    print(f"姓名: {name}, 年龄: {age}")

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

姓名: John, 年龄: 25
姓名: Eve, 年龄: 28

在这个例子中，使用了csv.DictReader来读取CSV文件，并将每一行数据转换为一个字典。然后我们遍历每一行，检查是否满足查询条件（居住城市为 "New York"），如果满足，则将姓名和年龄信息添加到extracted_data列表中。最后，我们输出提取到的特定数据。

这样，就可以根据需要从CSV文件中提取特定数据。注意，在实际应用中，可能需要加入一些错误处理逻辑，比如处理文件不存在或者特定字段不存在的情况。