問題描述

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

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

問題解答

你需要先讀文件，然后結(jié)果放到數(shù)組再對數(shù)組的值進(jì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內(nèi)容為

test1,output1test2,output2test3,output3

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