python實現vlookup,查找兩列中的相同數據並進行標記

#python實現vlookup,查找兩列中的相同數據並進行標記
import xlrd
import xlwt
import openpyxl

#打開工作簿
wb = openpyxl.load_workbook(r'C:\Users\TestBird\Desktop\Python\file\Test3.xlsx')
#選取sheet
sh = wb['Sheet1']
#定義兩個用於存放數據的list
list1 = []
list2 = []
#將excel的兩列存入list中
for data in list(sh.rows)[0:]: #如果需要去掉第一行的表頭就從1開始

    list1.append(data[0].value) #將第一列數據存入list1
    list2.append(data[1].value) #將第一列數據存入list2
for i in range(len(list1)):
    for j in range(len(list2)):
        if list1[i] == list2[j]:
            #將相同數據寫入第三列，寫在和第一列數據相同的位置
            sh.cell(row=i+1, column=3, value=list1[i])
#保存數據，關閉excel
wb.save(r'C:\Users\TestBird\Desktop\Python\file\Test3.xlsx')