Python(xlrd、xlwt模块)操

2020-01-17 12:53:27 浏览数 (1)

一、前言

关于Python的xlrd、xlwt模块的使用,推介另一位博客主的博文:https://www.cnblogs.com/zhoujie/p/python18.html

这篇里面有详细介绍这两个模块的基本用法。

以下是关于我运用xlrd、xlwt模块的一个实例。需求如下:

需求是用宏去做的,但是因为时间比较紧急,我用了1天去“研究”怎么用宏去写,发现作为一个VBA入门者,比较难短时间学习并解决这个问题,因为VBA的可读性比较差的缘故吧。

于是我选择用Python去实现。


二、主体

这个需求还算比较简单,主要是分为“读”和“写”两部分。

(一)读取部分

 从Excel文件“测试题.xls”里面的“表格数据1”,“表格数据2”,“数据透视表”三个sheet中提取区域和各区域的店铺,并要求同一区域内的店铺名称不重复。

需求也很简单,就是“区域”和“店铺名称”两个元素进行去重和读取。

1.读取思路

1.1读取范围

读取范围主要是从“表格数据1”,“表格数据2”,“数据透视表”三个sheet中提取区域和各区域的店铺,如下:

三个表都是从A1,B1或者A2,B2开始,但是表的末尾有些其他文字注释,于是我用的判断条件设为是否中文和是否为空值。

代码语言:javascript复制
#构造一个函数判断是否中文
def is_Chinese(word):
    for ch in word:
        if 'u4e00' <= ch <= 'u9fff':
            return True
    return False
1.2去重判断

读取到的数据需要一个临时的“容器”,打算利用单个区域和店铺放到一个元组里面,然后把所有的元组放到一个列表里面。

然后将每个新元组和列表里面的元组对比,看是否已经存在于列表中,从而达到去重的目的。

代码语言:javascript复制
def read_excel():
    # 打开文件
    workbook = xlrd.open_workbook(r'D:安装包测试题.xls')

    #写一个循环体,筛选出不重复的区域和店铺
    #将涉及到区域和店铺的三个sheet中,不重复的区域和店铺名称写入元组内并存于一个列表内
    sheet_name = ['表格数据1','表格数据2','数据透视表']
    tup1 = []
    for j in range(3):
        sheet_source = workbook.sheet_by_name(sheet_name[j])
        nrows = sheet_source.nrows
        for i in range(nrows):
            if is_Chinese(sheet_source.cell(i,0).value) == False and sheet_source.cell(i,0).value != "":
                a = (sheet_source.cell(i,0).value,sheet_source.cell(i,1).value)
                if a not in tup1 :
                    tup1.append(a)
                else:
                    pass
            else:
                 pass

 读取完数据,按照它的需求,是要弹出一个提示框的

代码语言:javascript复制
#python弹出窗口,提示“读取完成!”
    msg.showinfo("Excel_Reading","读取已经完成!")
1.3统计店铺数量

“容器”tup1列表里面,放的是原始的数据,需要写个循环,去统计不同区域内的店铺数量。

代码语言:javascript复制
    #利用将元组转为字典,并统计各个区域的店铺数量
    dict1 = {}
    for i in tup1:
        if i[0] not in dict1.keys():
            dict1[i[0]] = 1
        else:
            dict1[i[0]]  = 1

 读取部分完成。完整代码如下:

代码语言:javascript复制
import xlrd
import xlwt
import tkinter.messagebox as msg

def is_Chinese(word):
    for ch in word:
        if 'u4e00' <= ch <= 'u9fff':
            return True
    return False

#读取数据
def read_excel():
    # 打开文件
    workbook = xlrd.open_workbook(r'D:安装包测试题.xls')

    #写一个循环体,筛选出不重复的区域和店铺
    #将涉及到区域和店铺的三个sheet中,不重复的区域和店铺名称写入元组内并存于一个列表内
    sheet_name = ['表格数据1','表格数据2','数据透视表']
    tup1 = []
    for j in range(3):
        sheet_source = workbook.sheet_by_name(sheet_name[j])
        nrows = sheet_source.nrows
        for i in range(nrows):
            if is_Chinese(sheet_source.cell(i,0).value) == False and sheet_source.cell(i,0).value != "":
                a = (sheet_source.cell(i,0).value,sheet_source.cell(i,1).value)
                if a not in tup1 :
                    tup1.append(a)
                else:
                    pass
            else:
                 pass

    #python弹出窗口,提示“读取完成!”
    msg.showinfo("Excel_Reading","读取已经完成!")

    #利用将元组转为字典,并统计各个区域的店铺数量
    dict1 = {}
    for i in tup1:
        if i[0] not in dict1.keys():
            dict1[i[0]] = 1
        else:
            dict1[i[0]]  = 1
    return (dict1)

 (二)写入部分

需要新建一个Excel,创建一个叫“总表”的sheet,写入标题、表头,以及根据上述的read_excel函数返回的字典,写入内容。

1、样式部分

因为写入函数.write()里面有多个参数是用来设定你写入内容的样式的,所以这里做一个函数,把需要设定的参数做一个封包。

代码语言:javascript复制
def set_style(height,bold=False):
    style = xlwt.XFStyle()  # 初始化样式
    
    font = xlwt.Font()  # 为样式创建字体
    font.name = 'Times New Roman' 
    font.bold = bold
    font.color_index = 4
    font.height = height
    style.font = font

    alignment = xlwt.Alignment()# 为样式创建居中方式
    alignment.horz = xlwt.Alignment.HORZ_CENTER
    style.alignment = alignment

    
    borders = xlwt.Borders()  # 为样式创建边框
    borders.left = xlwt.Borders.MEDIUM  
    borders.right = xlwt.Borders.MEDIUM  
    borders.top = xlwt.Borders.MEDIUM  
    borders.bottom = xlwt.Borders.MEDIUM  

    borders.left_colour = 0x40 # 边框上色
    borders.right_colour = 0x40
    borders.top_colour = 0x40
    borders.bottom_colour = 0x40
    style.borders = borders

    return style

2、写入excel

代码语言:javascript复制
def write_excel(**dd): #两个**代表输入一个字典作为参数
    f = xlwt.Workbook() #创建工作簿

    '''
    创建第一个sheet:
        sheet1
    '''
    sheet1 = f.add_sheet(u'总表',cell_overwrite_ok=True) #创建sheet

    sheet1.col(1).width = 256 * 20 #调整列宽,256是一个固定的单位

    row1 = [u'区域',u'店铺数量(家)']

    #生成标题
    sheet1.write_merge(0,0,0,1,u'总表',set_style(300,True))

    #生成第二行表头
    for i in range(0,len(row1)):
        sheet1.write(1,i,row1[i],set_style(220,True))

    #写入数据
    i = 2
    for a,b in dd.items():
        if b>=10: #将店铺数量大于10的数据,写入Excel
            sheet1.write(i,0,a,set_style(220))
            sheet1.write(i,1,b,set_style(220))
            i  =1
        else:
            pass

    f.save('总表.xlsx') #保存文件,文件会保存在此Python脚本所在的文件夹内。

最后执行:

代码语言:javascript复制
write_excel(**read_excel())

效果图:


总结

应该说Python的xlrd、xlwt模块对于Excel的数据读取和写入非常简易方便。但是在使用xlwt时,存在一个问题,就是它无法直接对现有的Excel工作表进行写入,只能新开一个Excel。或者将现有Excel复制一个副本,另存为。

0 人点赞