python 匹配數據輸出數據

Question

我現在有一個list，裡面的內容是re=['test1','test2','test3'],還有一個csv或xlsx或txt,裡面有兩列資料

#我想達到的效果是如果re的元素等於檔案中的A列，這輸出這個檔案的AB列的那一行數據，這個具體該怎麼實現呢？

Answer 1

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

Answer 2

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