使用Python封裝excel操作指南

from openpyxl import load_workbook
from openpyxl.worksheet.worksheet import Worksheet


class ExcelHandler():
 '''
 操作Excel
 '''

 def __init__(self, file):
  '''初始化函數'''
  self.file = file

 def open_sheet(self, sheet_name) -> Worksheet:
  '''打開表單'''
  wb = load_workbook(self.file)
  sheet = wb[sheet_name]
  return sheet

 def read_header(self, sheet_name):
  '''獲取表單的表頭'''
  sheet = self.open_sheet(sheet_name)
  headers = []
  for i in sheet[1]:
   headers.append(i.value)
  return headers

 def read_rows(self,sheet_name):
  '''
  讀取除表頭外所有數據（除第一行外的所有數據）
  返回的內容是一個二維列表，若想獲取每一行的數據，可使用for循環或*解包
  '''
  sheet = self.open_sheet(sheet_name)
  rows = list(sheet.rows)[1:]

  data = []
  for row in rows:
   row_data = []
   for cell in row:
    row_data.append(cell.value)
   data.append(row_data)

  return data

 def read_key_value(self,sheet_name):
  '''
  獲取所有數據，且將表頭中的內容與數據結合展示（以字典的形式）
  如：[
  {'序號':1,'會員卡號': '680021685898','機場名稱':'上海機場'},
  {'序號':2,'會員卡號': '680021685899','機場名稱':'廣州機場'}
  ]
  '''
  sheet = self.open_sheet(sheet_name)
  rows = list(sheet.rows)

  # 獲取標題
  data = []
  for row in rows[1:]:
   rwo_data = []
   for cell in row:
    rwo_data.append(cell.value)
    # 列表轉換成字典，與表頭裡的內容使用zip函數進行打包
   data_dict = dict(zip(self.read_header(sheet_name),rwo_data))
   data.append(data_dict)
  return data


 @staticmethod
 def write_change(file,sheet_name,row,column,data):
  '''寫入Excel數據'''
  wb = load_workbook(file)
  sheet = wb[sheet_name]

  # 修改單元格
  sheet.cell(row,column).value = data
  # 保存
  wb.save(file)
  # 關閉
  wb.close()

import os

# ExcelDemo 所在的絕對路徑
base_dir = os.path.split(os.path.split(os.path.abspath(__file__))[0])[0]

# excel 文件放置目錄
excel_dir = os.path.join(base_dir,'excedir')

import os

from ExcelDemo.config.excel_handler import ExcelHandler
from ExcelDemo.config import dir_config

excelfile = os.path.join(dir_config.excel_dir,'航班信息.xlsx')


if __name__ == '__main__':
 excel = ExcelHandler(excelfile)
 header = excel.read_header('Sheet1')
 data = excel.read_rows('Sheet1')
 data2 = excel.read_key_value('Sheet1')

 print(header)
 '''
 ['序號', '會員卡號', '機場', '航班日期', '航班號']

 '''


 print('**********************************')
 print(data)
 '''
 [
 [1, '680021685898', '西安南航', 20200503, 'CZ6754'],
 [2, '189000177074', '（新疆）莎車機場', 20200603, 'CZ6880'], 
 [3, '480005387697', '新疆南航', 20200612, 'CZ5390'], 
 [4, '380025990156', '西安南航', 20200619, 'CZ6622']
 ]
 
 '''



 print('**********************************')
 print(data2)
 '''
 [
 {'序號': 1, '會員卡號': '680021685898', '機場': '西安南航', '航班日期':20200503, '航班號': 'CZ6754'},
 {'序號': 2, '會員卡號': '189000177074', '機場': '（新疆）莎車機場', '航班日期': 20200603, '航班號': 'CZ6880'}, 
 {'序號': 3, '會員卡號': '480005387697', '機場': '（新疆）庫車', '航班日期': 20200612, '航班號': 'CZ5390'},
 {'序號': 4, '會員卡號': '380025990156', '機場': '西安南航', '航班日期': 20200619, '航班號': 'CZ6622'}
 ]
 
 '''

 excel.write_change(excelfile,'Sheet1',6,5,'新增內容')
 excel.write_change(excelfile,'Sheet1',2,3,'修改內容')