欢迎来到代码驿站!

Python代码

当前位置:首页 > 软件编程 > Python代码

Python实现提取Excel指定关键词的行数据

时间:2022-09-27 11:11:24|栏目:Python代码|点击:

一、需求描述

1.图片展示

从如图所示的数据中提取含有"python"、"ubuntu"关键词的所有行数据,其它的不提取:

备注: 关键词和数据行列数可自定义!!!

提取前:

提取后:

2.提取方法

代码如下(示例):

import xlrd
import xlwt

data = xlrd.open_workbook(r'shuju.xlsx')
rtable = data.sheets()[0]
wbook = xlwt.Workbook(encoding='utf-8',style_compression = 0)
wtable = wbook.add_sheet('sheet1',cell_overwrite_ok = True)

count = 0
keyword = ('python')
keyword1 = ('ubuntu')  #可添加多个关键词
for i in range(0,40):  #区域按数据包含的行数进行填写,过多会显示超出范围(out of range)
    if rtable.cell(i,2).value == keyword or rtable.cell(i,3).value == keyword or rtable.cell(i,4).value == keyword or rtable.cell(i,5).value == keyword or rtable.cell(i,2).value == keyword1 or rtable.cell(i,3).value == keyword1 or rtable.cell(i,4).value == keyword1 or rtable.cell(i,5).value == keyword1:
        for j in range(0,5):
            wtable.write(i,j,rtable.row_values(i)[j])
        count += 1
print (count)
wbook.save(r'medicaldata.xls')

缺点:需要手动删除空白,容易出现超出范围错误!!

二、python提取第二版

1.图片展示

提取前:

提取后:

2.提取方法

代码如下(示例):

import os
import xlwt
import xlrd
from openpyxl import load_workbook

##目的文件夹
dirpath=r'E:\py\python3.7\test\test89tiqu'
keyword='python'

##遍历函数
def files(dirpath, suffix=['.xls', 'xlsx']):
    for root ,dirs ,files in os.walk(dirpath):
        for name in files:
            if name.split('.')[-1] in suffix:
                yield os.path.join(root, name)

if __name__ == '__main__':

    jieguo = xlwt.Workbook(encoding="ascii")  #生成excel
    wsheet = jieguo.add_sheet('sheet name') #生成sheet    
    y=0 #生成的excel的行计数
    try:
        file_list = files(dirpath)
        for filename in file_list:
            workbook = xlrd.open_workbook(filename) #读取源excel文件
            print(filename)
            sheetnum=workbook.nsheets  #获取源文件sheet数目
            for m in range(0,sheetnum):
                sheet = workbook.sheet_by_index(m) #读取源excel文件第m个sheet的内容
                nrowsnum=sheet.nrows  #获取该sheet的行数
                for i in range(0,nrowsnum):
                    date=sheet.row(i) #获取该sheet第i行的内容
                    for n in range(0,len(date)):
                        aaa=str(date[n]) #把该行第n个单元格转化为字符串,目的是下一步的关键字比对
                        print(aaa)
                        if aaa.find(keyword)>0: #进行关键字比对,包含关键字返回1,否则返回0
                            y=y+1
                            for j in range(len(date)):
                                wsheet.write(y,j,sheet.cell_value(i,j)) #该行包含关键字,则把它所有单元格依次写入入新生成的excel的第y行
        jieguo.save('jieguo.xls') #保存新生成的Excel
    except Exception as e:
        print(e)
                     
    jieguo.save('jieguo.xls') #保存新生成的Excel        

上一篇:python爬虫竟然被小伙用来算命

栏    目:Python代码

下一篇:python中 conda 虚拟环境管理和jupyter内核管理

本文标题:Python实现提取Excel指定关键词的行数据

本文地址:http://www.codeinn.net/misctech/214828.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有