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'
    )