python 匹配数据输出数据

Question

我现在有一个list，里面的内容为re=['test1','test2','test3'],还有一个csv或者xlsx或者txt,里面有两列数据

我想达到的效果是如果re的元素等于文件中的A列，这输出这个文件的AB列的那一行数据，这个具体应该怎么实现呢？

黄舟 · Answer

你需要先读文件，然后结果放到数组
再对数组的值进行查找

欧阳克 · Answer

import csv
with open("lookup.csv") as f:
    reader = csv.reader(f, delimiter=',')
    dict_lookup = {r[0]:r[1] for r in reader}
print(dict_lookup) 
print(dict_lookup['test2'])

输出

{'test1': 'output1', 'test2': 'output2', 'test3': 'output3'}
output2

若lookup.csv内容为

test1,output1
test2,output2
test3,output3

这是很基本的csv模块使用操作，建议看一下