Python脚本:从CSV文件匹配ID并打印对应文本
下面是一个示例的Python脚本,可以实现从键盘读取行,将内容与CSV文件中的“id”列进行比较,并打印对应的“text”列的内容:
import csv
# 读取CSV文件
with open('results-20230616T1300.csv', 'r', encoding='utf-8') as csvfile:
reader = csv.DictReader(csvfile)
# 创建字典,存储id和text的对应关系
id_text_map = {row['id']: row['text'] for row in reader}
# 从键盘读取行,进行比较并打印结果
while True:
input_id = input("请输入id(输入exit退出):")
if input_id == "exit":
break
if input_id in id_text_map:
print(id_text_map[input_id])
else:
print("未找到匹配的id")
请将脚本放在与CSV文件相同的目录下,并将CSV文件命名为"results-20230616T1300.csv"。脚本将循环读取键盘输入的id,如果匹配到CSV文件中的id,则打印对应的text内容。如果输入"exit",则退出循环。如果未找到匹配的id,则打印"未找到匹配的id"。
原文地址: https://www.cveoy.top/t/topic/pV66 著作权归作者所有。请勿转载和采集!