Python模块--openpyxl

openpyxl 是一个第三方库，可以处理 xlsx 格式的 Excel 文件。
需要进行安装 pip install openpyxl

1. 读取Excel文件

读取文件–load_workbook

• load_workbook 是读取 Ecxel 文件的类

from openpyxl import load_workbook
# 默认可读写，若有需要可以指定write_only和read_only为True
wb = load_workbook('test.xlsx')

获取工作表–sheetnames

• sheetnames 可以获取所有 sheet 页名字的方法，以列表形式返回
• wb[sheet_name] 根据 sheet 页的名字获取

# 获得所有sheet的名称
print(wb.sheetnames)
# 根据sheet名字获得sheet
a_sheet = wb['Sheet1']
# 获得sheet名
print(a_sheet.title)
# 获得当前正在显示的sheet
print(wb.active)
# 输出
['Sheet1', 'Sheet2']
Sheet1
<Worksheet "Sheet2">

获取单元格值–sheet[A1]

# 获取某个单元格的值
b4 = sheet['B4']
# 分别返回
print(f'({b4.column}, {b4.row}) is {b4.value}') # 返回的数字就是int型
# 除了用下标的方式获得，还可以用cell函数, 换成数字，这个表示B2
b4_too = sheet.cell(row=4, column=2)
print(b4_too.value)
# 输出
B 4 test

获得最大行\列–sheet.max_row/column

# 获得最大列和最大行
print(sheet.max_row)
print(sheet.max_column)

获取行和列–sheet.rows/columns

sheet.rows 为生成器, 里面是每一行的数据，每一行又由一个tuple包裹。
sheet.columns 类似，不过里面是每个tuple是每一列的单元格。

# 因为按行，所以返回A1, B1, C1这样的顺序
for row in sheet.rows:
  for cell in row:
    print(cell.value)
# A1, A2, A3这样的顺序
for column in sheet.columns:
  for cell in column:
    print(cell.value)

上面的代码就可以获得所有单元格的数据。如果要获得某行的数据呢？给其一个索引就行了，因为 sheet.rows是生成器类型，不能使用索引，转换成list之后再使用索引， list(sheet.rows)[2] 这样就获取到第二行的 tuple 对象。

for cell in list(sheet.rows)[2]:
  print(cell.value)

获取区间单元格

像使用切片那样， sheet['A1':'B3'] 返回一个 tuple，该元组内部还是元组，由每行的单元格构成一个元组。

for row_cell in sheet['A1':'B3']:
  for cell in row_cell:
    print(cell)
    
for cell in sheet['A1':'B3']:
  print(cell)
# 输出
(<Cell pythontab.A1>, <Cell pythontab.B1>)
(<Cell pythontab.A2>, <Cell pythontab.B2>)
(<Cell pythontab.A3>, <Cell pythontab.B3>)

根据字母获得列号，根据列号返回字母
需要导入， 这两个函数存在于 openpyxl.utils

from openpyxl.utils import get_column_letter, column_index_from_string
# 根据列的数字返回字母
print(get_column_letter(2)) # B
# 根据字母返回列的数字
print(column_index_from_string('D')) # 4

2. 写入Excel文件

创建文件对象–WorkBook

需要导入WorkBook

from openpyxl import Workbook
wb = Workbook()   # 新建了一个工作簿，若要指定只写模式，可以指定参数write_only=True。一般默认的可写可读模式就可以了。
# 直接赋值就可以改工作表的名称
sheet.title = 'Sheet1'
# 新建一个工作表，可以指定索引使其在工作簿中的位置指定
w2 = wb.create_sheet(title='S2', index=1) # 表示新建一个sheet页，位置为第二个

单元格赋值

直接以键值对的方式赋值

# 直接给单元格赋值就行
w2['A1'] = 'hello'

# A3处写入平均值
w2['A5'] = '=AVERAGE(A2:A4)'

但是如果是读取的时候需要加上 data_only=True 这样读到 A5 返回的就是数字，如果不加这个参数，返回的将是公式本身 =AVERAGE(A2:A4)

append 
可以一次添加多行数据，从第一行空白行开始（下面都是空白行）写入。

w3 = wb.create_sheet(title="S3")
# 添加一行
row = [1 ,2, 3, 4, 5]
sheet.append(row)
# 添加多行
rows = [
  ['Number', 'data1', 'data2'],
  [2, 40, 30],
  [3, 40, 25],
  [4, 50, 30],
  [5, 30, 10],
  [6, 25, 5],
  [7, 50, 10],
]

由于append函数只能按行写入。矩阵转置可以帮助我们实现按列写入

rows = [
  ['Number', 'data1', 'data2'],
  [2, 40],
  [3, 40, 25],
  [4, 50, 30],
  [5, 30, 10],
  [6, 25, 5],
  [7, 50, 10],
]
print(list(zip(*rows)))
# 输出
[('Number', 2, 3, 4, 5, 6, 7), ('data1', 40, 40, 50, 30, 25, 50)]

for row in list(zip(*rows)):
	w3.append(row)

添加图片–add_image

from openpyxl import Workbook
from openpyxl.drawing.image import Image

wb = Workbook()
ws = wb.active
ws['A1'] = 'img here'
img = Image('demo.png')
ws.add_image(img, 'A1')		# 添加图片至A1单元格

3. 保存文件

所有的操作结束后，保存文件。指定路径和文件名，后缀名为 xlsx

wb.save('test.xlsx')

设置单元格风格–Style

from openpyxl.styles import Font, colors, Alignment

字体

bold_itatic_24_font = Font(name='等线', size=24, italic=True, color=colors.RED, bold=True)
w2['A1'].font = bold_itatic_24_font

上面的代码指定了等线24号加粗斜体，字体颜色红色。直接使用cell的 font 属性，将 Font 对象赋值给它。

对齐方式

也是直接使用 cell 的属性 aligment ，这里指定垂直居中和水平居中。除了 center ，还可以使用 right、left 等等参数。

# 设置B1中的数据垂直居中和水平居中
w2['B1'].alignment = Alignment(horizontal='center', vertical='center')

设置行高和列宽

有时候数据太长显示不完，就需要拉长拉高单元格。

# 第2行行高
w2.row_dimensions[2].height = 40
# C列列宽
w2.column_dimensions['C'].width = 30

合并和拆分单元格

合并单元格，即以合并区域的左上角的那个单元格为基准，覆盖其他单元格使之称为一个大的单元格。
相反，拆分单元格后将这个大单元格的值返回到原来的左上角位置。

# 合并单元格， 往左上角写入数据即可
w2.merge_cells('B1:G1') # 合并一行中的几个单元格
w2.merge_cells('A1:C3') # 合并一个矩形区域中的单元格

合并后只可以往左上角写入数据，也就是区间中:左边的坐标。
如果这些要合并的单元格都有数据，只会保留左上角的数据，其他则丢弃。换句话说若合并前不是在左上角写入数据，合并后单元格中不会有数据。
以下是拆分单元格的代码。拆分后，值回到A1位置。

w2.unmerge_cells('A1:C3')

官方文档