04-openpyxl处理Excel

1. 创建Excel的工作簿

首先需要导入openpyxl

from openpyxl import Workbook
wb = Workbook()
ws = wb.active # 得到默认的工作簿， 工作簿默认是从0进行编号的
ws1 = wb.create_sheet('HmSheet1') # 创建一个新的工作簿，并位于最后
ws2 = wb.create_sheet('HmSheet2',0)  # 创建一个新的工作簿，并位于最前
ws3 = wb.create_sheet('HmSheet3',-1) # 创建一个新的工作簿 并位于倒数第二个位置
ws.title = 'HmSheet0' # 修改Sheet页的标题
ws0.sheet_properties.tabColor = '1072BA' #修改Sheet页的标题的背景色，默认为白色

获取某一个Sheet

ws = wb.get_index(0)
ws = wb.get_sheet_by_name('HM测试1')
ws = wb['HM测试1']

获取所有Sheet名称

print(wb.sheetnames)

遍历Sheet

for sheet in wb:
	print(sheet.title)

复制Sheet

 ws0 = wb.active
 ws4 = wb.copy_worksheet(ws0)

2. 读写数据

# 数据写入
ws1['A4'] = '测试文件'
ws1.cell(row=4, column=2, value=10)
ws1.cell(row=4, column=2, value="{0}".format(get_column_letter(2))) # get_column_letter(2)是根据数字得到字母， column_index_from_string('D') 根据字母得到数字， 这两个方法需要导入 from openpyxl.utils import get_column_letter, column_index_from_string
ws['A3'] = datetime.datetime.now().strftime("%Y-%m-%d") 
ws['A1'] = datetime.datetime(2010, 7, 21)
ws['A1'].number_format
ws["A1"] = "=SUM(1, 1)"
# 数据读取
print(ws1['A4'].value)

获取范围数据

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

cell_range = ws0['A1':'C2'] # 访问A1 到 C2
for rows in cell_range:
	for row in rows:
    	print(row.value)

colC = ws['C']	# 获取C列数据
col_range = ws['C:D'] # 获取C列到D列数据
row10 = ws[10]  # 获取第10行数据
row_range = ws[5:10] # 获取第5行到第10行数据

for row in ws.iter_rows(min_row=1, max_col=3, max_row=2):
	for cell in row:
		print(cell)

for col in ws.iter_cols(min_row=1, max_col=3, max_row=2):
	for cell in col:
		print(cell)

获取单元格中的值

for row in ws.values:
   for value in row:
     print(value)

使用Worksheet.iter_rows()并Worksheet.iter_cols()可以采取的values_only参数，只返回单元格的值：

for row in ws.iter_rows(min_row=1, max_col=3, max_row=2, values_only=True):
	print(row)

删除工作表

# 方式一
wb.remove(sheet)
# 方式二
del wb[sheet]
⑥ 矩阵置换（行 → 列）

矩阵置换（行 → 列）

rows = [
    ['Number', 'data1', 'data2'],
    [2, 40, 30],
    [3, 40, 25],
    [4, 50, 30],
    [5, 30, 10],
    [6, 25, 5],
    [7, 50, 10]]

list(zip(*rows))

# out
[('Number', 2, 3, 4, 5, 6, 7),
 ('data1', 40, 40, 50, 30, 25, 50),
 ('data2', 30, 25, 30, 10, 5, 10)]

# 注意 方法会舍弃缺少数据的列(行)
rows = [
    ['Number', 'data1', 'data2'],
    [2, 40      ],    # 这里少一个数据
    [3, 40, 25],
    [4, 50, 30],
    [5, 30, 10],
    [6, 25, 5],
    [7, 50, 10],
]
# out
[('Number', 2, 3, 4, 5, 6, 7), ('data1', 40, 40, 50, 30, 25, 50)]

3. 保存文件

保存工作簿的最简单和最安全的方式是通过使用Workbook.save()所述的方法Workbook对象：

wb = Workbook()
wb.save('balances.xlsx')

4. 保存为流

我们在使用Web框架时， 文件以流的形式返回

from tempfile import NamedTemporaryFile
from openpyxl import Workbook

wb = Workbook()
with NamedTemporaryFile() as tmp:
        wb.save(tmp.name)
        tmp.seek(0)
        stream = tmp.read()

您可以指定属性template=True，对工作簿保存为模板：

wb = load_workbook('document.xlsx')
wb.template = True
wb.save('document_template.xltx')

或将此属性设置为假（默认），将其保存为文件：

wb = load_workbook('document_template.xltx')
wb.template = False
wb.save('document.xlsx', as_template=False)

5. 加载文件

from openpyxl import load_workbook
wb2 = load_workbook('test.xlsx')
print(wb2.sheetnames)

['Sheet2', 'New Title', 'Sheet1']

6. 设置单元格风格

需要导入的类

from openpyxl.styles import Font, colors, Alignment

字体

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

bold_itatic_24_font = Font(name='等线', size=24, italic=True, color=colors.RED, bold=True)

sheet['A1'].font = bold_itatic_24_font

对齐方式

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

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

设置行高和列宽

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

7. 合并和拆分单元格

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

# 合并单元格， 往左上角写入数据即可
sheet.merge_cells('B1:G1') # 合并一行中的几个单元格
sheet.unmerge_cells('B1:G1')  # 拆分

ws.merge_cells(start_row=2, start_column=1, end_row=4, end_column=4) # 合并
ws.unmerge_cells(start_row=2, start_column=1, end_row=4, end_column=4) # 拆分

sheet.merge_cells('A1:C3') # 合并一个矩形区域中的单元格
sheet.unmerge_cells('A1:C3')  # 拆分

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

sheet.unmerge_cells('A1:C3')

7. 插入图片

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

wb = Workbook()
ws = wb.active
ws['A1'] = 'You should see three logos below'
# create an image
img = Image('logo.png')
# add to worksheet and anchor next to cells
ws.add_image(img, 'A1')
wb.save('logo.xlsx')

8. 隐藏列

import openpyxl
wb = openpyxl.Workbook()
ws = wb.create_sheet()
ws.column_dimensions.group('A','D', hidden=True)
# ws.row_dimensions.group(1,10, hidden=True) # 报错
wb.save('group.xlsx')