問題描述

我現(xiàn)在有一個list，里面的內容為re=[’test1’,’test2’,’test3’],還有一個csv或者xlsx或者txt,里面有兩列數據

我想達到的效果是如果re的元素等于文件中的A列，這輸出這個文件的AB列的那一行數據，這個具體應該怎么實現(xiàn)呢？

問題解答

你需要先讀文件，然后結果放到數組再對數組的值進行查找

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模塊使用操作，建議看一下