經常使用模塊 - openpyxl模塊

1、簡介

xlrd/xlwt

         主要是針對Office 2003或更早版本的XLS文件格式

         缺點：不支持XLSX文件格式

OpenPyXL

         能讀能寫能修改

         缺點：不支持XLS

Microsoft Excel API

         強大無極限

         缺點：速度慢

2、基本用法

首先介紹下Excel的一些基本概念，Workbook至關因而一個文件，WorkSheet就是文件裏面的每一個具體的表，好比新建Excel文件裏面的「Sheet1」，一個Workbook裏面有一個或多個WorkSheet。

2.1操做Workbook對象

獲取Workbook對象的方法有兩種，一種是建立一個新的，一種是導入一個已經存在的。

2.1.1獲取Workbook對象

方法1：建立Workbook

# 導入模塊
from openpyxl import Workbook
# 建立一個Workbook
wb = Workbook() // 默認生成一個名爲'Sheet'的WorkSheet

方法2：導入Workbook

# 導入模塊
from openpyxl import load_workbook
# 導入一個Workbook
wb = load_workbook(filename='test.xlsx')

2.1.2Workbook屬性

sheetnames： 返回全部WorkSheet的名字列表，類型爲list
worksheets： 返回全部WorkSheet的列表，類型爲list
active： 返回當前默認選中的WorkSheet

2.1.3Workbook方法

get_sheet_names()： 同sheetnames
get_active_sheet()： 同active屬性
get_sheet_by_name(name)： 根據名稱獲取WorkSheet
remove(worksheet)： 刪除一個WorkSheet，注意是WorkSheet對象，不是名字
save(filename)： 保存到文件，記住有寫入操做記得保存。

2.2操做WorkSheet

2.2.1獲取WorkSheet對象

# 獲取默認打開的(active)的WorkSheet
ws1 = wb.active
# 建立一個WorkSheet
ws2 = wb.create_sheet() # 可傳title和index兩個參數,不傳生成的WorkSheet名在'Sheet'後面遞增長數字
# 經過名稱獲取WorkSheet
ws3=wb['Sheet1']

2.2.2WorkSheet屬性

rows： 返回全部有效數據行，有數據時類型爲generator，無數據時爲tuple
columns： 返回全部有效數據列，類型同rows
max_column： 有效數據最大列
max_row： 有效數據最大行
min_column： 有效數據最小列，起始爲1
min_row： 有效數據最大行，起始爲1
values： 返回全部單元格的值的列表，類型爲tuple
title： WorkSheet的名稱

2.2.3WorkSheet方法

cell(coordinate=None, row=None, column=None, value=None)： 獲取指定單元格或設置單元格的值，具體使用在cell下面介紹

2.3操做Cell

2.3.1獲取Cell對象

# 使用WorkSheet的Cell方法
c1 = ws.cell('A1')
c2 = ws.cell(row=1, column=1)  # 獲取A1單元格
# 經過座標獲取Cell
c3 = ws['A1']
# 獲取多個
c3 = ws['A1:E5']  # 返回多行數據,類型爲tuple

2.3.2設置Cell的值

# 直接使用WorkSheet的cell方法設置
ws.cell(row=1, column=1, value=10)
# 設置Cell對象value屬性
c1 = ws.cell('A1')
c1.value = 100

2.3.3Cell屬性

column： 所在列，起始爲1
row： 所在行，起始爲1
coordinate：  所在座標，如'A1'
parent： 所屬的WorkSheet
value： 單元格的值

2.3.4Cell方法

offset(row=0, column=0)： 偏移

3、使用實例

#! /usr/bin/env python3
# -*- coding:utf-8 -*-

# Author   : mayi
# Blog     : http://www.cnblogs.com/mayi0312/
# Date     : 2019/04/25
# Name     : my_excel_model
# Software : PyCharm
# Note     : 功能：使用openpyxl模塊處理Excel文件

import os
from openpyxl import load_workbook
from openpyxl import Workbook

def getExcelConOpenpyxl(file_name, book_name=None, data_type="list", index=1):
    """
    利用openpyxl模塊獲取Excel表中的內容
    :param file_name: Excel表文件名
    :param book_name: Sheet表名
    :param data_type: 列表中存儲每行內容的方式（list或dict），默認爲列表
    :param index: sheet的序號或者名字，默認處理第一個sheet
    :return: Excel表中的內容
    """
    # 打開一個xlsx文件
    wb = load_workbook(file_name)
    # 打開指定的sheet
    # sheet = wb.get_sheet_by_name("名單")
    # sheet = wb["名單"]
    if not book_name:
        sheet = wb.active
    else:
        sheet = wb.get_sheet_by_name(book_name)
    # 用於存儲Excel表內容的列表
    excel_con_list = []
    # 獲取全部內容
    line_list = sheet.rows
    # 第一行爲表頭
    field_list = next(line_list)
    for line in line_list:
        if data_type == "dict":
            # 字典
            temp_line_dict = {}
            for i, cell in enumerate(line):
                cell_key = field_list[i].value
                cell_value = cell.value
                temp_line_dict[cell_key] = cell_value
            excel_con_list.append(temp_line_dict)
        else:
            # 列表
            temp_line_list = []
            for i, cell in enumerate(line):
                cell_key = field_list[i].value
                cell_value = cell.value
                temp_line_list.append([cell_key, cell_value])
            excel_con_list.append(temp_line_list)

    # 關閉文檔
    wb.close()

    return excel_con_list

def setExcelConOpenpyxl(file_name, con_list, field_list=None, book_name=None, model=None):
    """
    利用openpyxl模塊將列表中的內容存儲至Excel表中
    :param file_name: Excel表文件名
    :param con_list: 須要保存的列表信息
    :param field_list: 表頭內容
    :param book_name: Sheet表名稱
    :param model: 模板文件名（默認沒有模板）
    :return: None
    """
    if model:
        # 若是有模板文件，則打開模板工做簿
        wb = load_workbook(model)
    else:
        # 不然，新建一個Excel工做簿
        wb = Workbook()
    # 激活工做表
    sheet = wb.active
    if not book_name:
        sheet.title = book_name
    if field_list:
        # 若是有設置表頭的話
        # 列數
        n_col = len(field_list)
        for i_col in range(n_col):
            sheet.cell(row=1, column=i_col + 1, value=field_list[i_col])
    # 行數
    n_row = len(con_list)
    for i in range(n_row):
        # 列數
        n_col = len(con_list[i])
        for j in range(n_col):
            sheet.cell(row=2 + i, column=j + 1, value=con_list[i][j])

    if not os.path.exists(os.path.dirname(file_name)):
        # 若是文件所在目錄不存在，則建立
        os.makedirs(os.path.dirname(file_name))

    # 文檔另存爲
    wb.save(file_name)
    # 關閉文檔
    wb.close()

# 入口函數
if __name__ == '__main__':
    filename = "test.xlsx"
    con_list = getExcelConOpenpyxl(filename)
    print(con_list)
    filename = os.path.abspath("test_w.xlsx")
    field_list = ["序號", "姓名", "性別", "年齡"]
    con_list = [
        [1, "張三", "男", "20"],
        [2, "李四", "女", "21"],
        [3, "王五", "男", "19"],
        [4, "麻六", "女", "24"],
        [5, "田七", "男", "23"],
    ]
    setExcelConOpenpyxl(filename, con_list, field_list, book_name="名單")

注意：

         ·Cell的row和column都是從1開始的

         ·文件操做完記得調用Workbook的save()方法

         ·最好記得調用Workbook的close()方法