Python提取CSV指定数据：筛选特定单品名称

本文将介绍如何使用Python从CSV文件中筛选出包含特定单品名称的行，并将结果保存到新的CSV文件中。

需求：

假设有一个名为'data.csv'的CSV文件，其中包含多列数据，其中一列为'单品名称'。我们需要筛选出'单品名称'列的值为以下内容的行：

白玉菇(袋)
菠菜
菠菜(份)
紫茄子(1)
紫茄子(2)

并将筛选结果保存到名为'data3.csv'的新文件中。

解决方案：

以下Python代码可以实现上述需求：

import csv

# 指定要筛选的单品名称
target_items = ['白玉菇(袋)', '菠菜', '菠菜(份)', '紫茄子(1)', '紫茄子(2)']

# 读取原始数据文件
with open('data.csv', 'r') as infile, open('data3.csv', 'w', newline='') as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)
    
    # 遍历每一行数据
    for row in reader:
        # 检查'单品名称'列是否包含在目标列表中
        if row[0] in target_items:
            # 如果包含，则将该行写入新文件
            writer.writerow(row)

代码说明：

首先，我们定义了一个名为target_items的列表，其中包含要筛选的单品名称。
使用with open()函数打开原始数据文件'data.csv'和新文件'data3.csv'。
使用csv.reader()和csv.writer()函数创建读取器和写入器对象。
使用循环遍历原始CSV文件中的每一行数据。
在循环内部，使用if语句检查当前行的第一个元素（即'单品名称'列）是否包含在target_items列表中。
如果包含，则使用writerow()函数将该行写入新的CSV文件'data3.csv'中。

使用方法：

将以上代码保存为Python文件，例如'filter_csv.py'。
将原始数据文件命名为'data.csv'，并将其与Python脚本文件放在同一目录下。
运行Python脚本文件python filter_csv.py。
脚本运行完成后，将在同一目录下生成一个名为'data3.csv'的新文件，其中包含筛选后的数据。

总结：

本文介绍了一种使用Python提取CSV指定数据的方法，通过筛选特定单品名称，我们可以快速从大量数据中获取所需信息。你可以根据实际需求修改代码中的目标列表和文件名，以适应不同的数据处理场景。