Python 程序:提取指定单品名称数据
以下是一个示例 Python 程序,它会读取名为 'data.csv' 的文件,并将符合条件的行(即包含指定单品名称的行)写入名为 'data3.csv' 的文件中。
import csv
# 打开输入文件
with open('data.csv', 'r', encoding='utf-8') as file:
reader = csv.reader(file)
# 创建输出文件
with open('data3.csv', 'w', encoding='utf-8', newline='') as outfile:
writer = csv.writer(outfile)
# 写入标题行
writer.writerow(next(reader))
# 遍历每一行数据
for row in reader:
# 判断单品名称是否符合条件
if row[1] in ['白玉菇(袋)', '菠菜', '菠菜(份)', '菜心', '虫草花(份)', '高瓜(1)', '高瓜(2)', '海鲜菇(包)', '红椒(2)',
'红莲藕带', '红薯尖', '洪湖藕带', '姜蒜小米椒组合装(小份)', '金针菇(盒)', '净藕(1)', '菱角', '螺丝椒',
'螺丝椒(份)', '木耳菜', '木耳菜(份)', '奶白菜', '七彩椒(2)', '青红杭椒组合装(份)', '青茄子(1)',
'青线椒(份)', '上海青', '双孢菇(盒)', '娃娃菜', '外地茼蒿', '芜湖青椒(1)', '西兰花', '西峡花菇(1)',
'鲜木耳(份)', '苋菜', '小米椒(份)', '小青菜(1)', '小皱皮(份)', '蟹味菇与白玉菇双拼(盒)', '野生粉藕',
'圆茄子(2)', '云南生菜', '云南生菜(份)', '云南油麦菜', '云南油麦菜(份)', '长线茄', '枝江青梗散花',
'竹叶菜', '紫茄子(1)', '紫茄子(2)']:
# 写入符合条件的行
writer.writerow(row)
请注意,此示例程序假设输入文件中的第一行是标题行,并将其复制到输出文件中。如果你的输入文件不包含标题行,请删除 writer.writerow(next(reader)) 这一行。
此外,你需要将示例程序中的 data.csv 和 data3.csv 替换为你实际使用的文件名。
原文地址: https://www.cveoy.top/t/topic/epk2 著作权归作者所有。请勿转载和采集!