python简单处理excel方法

 

python自带xlrd和xlwt模块用来处理excel，但总觉得xlwt模块用着别扭，于是按自己的习惯重新封装了一个

 

# coding=utf-8

import xlrd   # 读模块
import xlwt   # 写模块


def read_excel():
    """
        读取excel文件方法说明（此函数只是对xlrd用法的说明，看看就好。xlrd已经很好用了）
    :return:
    """
    # 打开excel文件
    data = xlrd.open_workbook('123.xls')

    # 获取文件内的表名
    print(data.sheet_names())

    # 获取指定表的两种方法
    sh = data.sheet_by_index(0)  # 索引法
    sh = data.sheet_by_name(u'Sheet1')  # 表名法

    # 获取表的行数
    print(sh.nrows)
    # 获取表的列数
    print(sh.ncols)

    # 获取第29行的数据(0算第一行)
    print(sh.row_values(28))
    # 获取第2列的数据(0算第一列)
    print(sh.col_values(1))
    # 获取第1行、第2列的数据(相当于坐标)
    print(sh.cell(0, 1).value)


def write_excel(sheet_name, row0, datas, file_name):
    """
        自己重新封装的写入excel表模块。以后要将数据写入excel直接调用此函数就行。此函数简单干净，使用方便。
    :param sheet_name: excel文件内的表名。例：Sheet1
    :param row0: 表内第一行的内容，为一个列表。例：[id, name, sex]
    :param datas: 实际要写入表内的数据，为一个列表，里面包含几个元素就写几行数据，其中每一个元素又为一个列表，里面对应这一行
    每列的值。例：[[1, '小龙女', 'girl'], [2, '杨过', 'boy'], [3, '周伯通', 'boy']]
    :param file_name: 最终生成的excel文件名。例：'test.xls'
    :return:
    """
    # 创建一个写对象实例
    f = xlwt.Workbook()

    # 创建一个叫sheet_name的表名
    sheet1 = f.add_sheet(sheet_name, cell_overwrite_ok=True)

    # 创建表sheet_name的第一行(即标题)
    for i in range(0, len(row0)):
        sheet1.write(0, i, row0[i])

    # 从第二行开始写数据，datas有多少元素就写多少行
    for row in range(1, len(datas) + 1):
        # 遍历每行的每一列，row0有多少元素就遍历多少列
        for col in range(len(row0)):
            # 有了上面row,col提供的行号和列号就可以精确匹配一个单元格了，下面就开始往单元格里写数据了
            # row：行号，col：列号，datas[row-1][col]：数据
            # 注：datas这个列表里一个元素为一行，由于是从第二行开始写的，所以想要取datas的第一个元素需row-1
            # 注：datas这个列表里一个元素(列表)里的元素对应这行每列的值。
            # 例:[1, '小龙女', 'girl'],1对应第1列的值,'小龙女'对应第2列的值,'girl'对应第三列的值
            sheet1.write(row, col, datas[row-1][col])

    f.save(file_name)
    print('已生成excel文件！')


if __name__ == '__main__':
    read_excel()
    write_excel(
        sheet_name='students',
        row0=['id', 'name', 'sex'],
        datas=[[1, '小龙女', 'girl'], [2, '杨过', 'boy'], [3, '周伯通', 'boy']],
        file_name='test.xls'
    )