python 匹配数据输出数据

问题描述

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

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

问题解答

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

import csvwith 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,output1test2,output2test3,output3

這是很基本的csv模塊使用操作，建議看一下