有两个文件,txt和excel
其中txt存放的是关键词,excel存放的是内容
txt存放关键词的形式是
成都
北京
天津
上海
重庆
excel存放的是一行一行的内容
能否将txt和excel进行匹配,提取出含有关键词的行,并存放到新的文件里?
有两个文件,txt和excel
其中txt存放的是关键词,excel存放的是内容
txt存放关键词的形式是
成都
北京
天津
上海
重庆
excel存放的是一行一行的内容
能否将txt和excel进行匹配,提取出含有关键词的行,并存放到新的文件里?
def get_key_list(txt_file):
try:
file_data = open(txt_file, 'r')
list1 = file_data.readlines()
key_list = [k.strip('\n') for k in list1]
return key_list
except:
pass
return []
def xls_select(xls_sr, xls_tg, key_list):
from openpyxl import Workbook
from openpyxl import load_workbook
# 只处理第一个工作表 , 数据在第一列
wb = load_workbook(xls_sr)
ws = wb[wb.sheetnames[0]]
wb_tg = Workbook()
sheet_tg = wb_tg.active
sheet_tg.title = "Data Select"
save_i = 0
for row in ws.rows:
for k in key_list:
if row[0].value.find(k) >= 0:
sheet_tg["A%d" % (save_i + 1)].value = row[0].value
save_i = save_i + 1
wb_tg.save(xls_tg)
wb.close()
wb_tg.close()
key_list = get_key_list('key.txt')
xls_select('X1.xlsx', 'X2.xlsx', key_list)
试试,基本能完成你要结果