fatesses
fatesses
采纳率98%
2020-02-25 14:39

python 匹配excel和txt文件,提取含有关键词的行

10
已采纳

有两个文件,txt和excel
其中txt存放的是关键词,excel存放的是内容

txt存放关键词的形式是

成都
北京
天津
上海
重庆

excel存放的是一行一行的内容
能否将txt和excel进行匹配,提取出含有关键词的行,并存放到新的文件里?

  • 点赞
  • 写回答
  • 关注问题
  • 收藏
  • 复制链接分享
  • 邀请回答

2条回答

  • seakingx 陈年椰子 1年前
     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)
    

    试试,基本能完成你要结果

    点赞 评论 复制链接分享
  • caozhy 回答这么多问题就耍赖把我的积分一笔勾销了 1年前

    用pandas读取excel或者txt(也就是csv)然后过滤。
    https://blog.csdn.net/qq_31112205/article/details/82702875

    点赞 评论 复制链接分享