教你如何用Python处理Excel表格

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

读取Excel文件

需要导入相关函数

from openpyxl import load_workbook
# 默认可读写，若有需要可以指定write_only和read_only为True
wb = load_workbook('pythontab.xlsx')
默认打开的文件为可读写，若有需要可以指定参数read_only为True。

获取工作表--Sheet

# 获得所有sheet的名称
print(wb.get_sheet_names())
# 根据sheet名字获得sheet
a_sheet = wb.get_sheet_by_name('Sheet1')
# 获得sheet名
print(a_sheet.title)
# 获得当前正在显示的sheet, 也可以用wb.get_active_sheet()
sheet = wb.active

获取单元格

# 获取某个单元格的值，观察excel发现也是先字母再数字的顺序，即先列再行
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)

b4.column 返回 B，b4.row 返回 4，value 是该单元格的值。另外，单元格还有一个属性坐标，像b4这样的单元格返回坐标B4。

获得行和列

获取行和列

sheet.rows 是一个生成器，其中包含每一行的数据，每一行都被一个元组包裹。

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(sheet.rows)[2] 这样就得到了 tuple 对象第二行。

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

如何获得任意区间的单元格？

可以使用range函数，如下写法，得到矩形区域内的所有单元格，A1为左上角，B3为右下角。注意范围是从1开始的，因为在openpyxl中，为了和Excel中的表达方式一致，不符合编程语言的习惯用0来表示第一个值。

for i in range(1, 4):
  for j in range(1, 3):
    print(sheet.cell(row=i, column=j))
     
# out
<Cell pythontab.A1>
<Cell pythontab.B1>
<Cell pythontab.A2>
<Cell pythontab.B2>
<Cell pythontab.A3>
<Cell pythontab.B3>

它也可以像切片一样使用。 sheet['A1':'B3'] 返回一个元组，里面也是一个元组，每行的单元格组成一个元组。

for row_cell in sheet['A1':'B3']:
  for cell in row_cell:
    print(cell)
     
for cell in sheet['A1':'B3']:
  print(cell)
# out
(<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

将数据写入Excel

工作表相关

需要导入WorkBook

from openpyxl import Workbook
wb = Workbook()

这会创建一个新的工作表（只是尚未保存）。

要指定只写模式，可以指定参数 write_only=True。一般情况下，默认的可读写模式就足够了。

print(wb.get_sheet_names()) # 提供一个默认名叫Sheet的表，office2016下新建提供默认Sheet1
# 直接赋值就可以改工作表的名称
sheet.title = 'Sheet1'
# 新建一个工作表，可以指定索引，适当安排其在工作簿中的位置
wb.create_sheet('Data', index=1) # 被安排到第二个工作表，index=0就是第一个位置
# 删除某个工作表
wb.remove(sheet)
del wb[sheet]

写入单元格

还可以使用公式

# 直接给单元格赋值就行
sheet['A1'] = 'good'
# B9处写入平均值
sheet['B9'] = '=AVERAGE(B2:B8)'

但是如果需要在读取的时候加上data_only=True，B9返回的数字就是数字。如果不添加此参数，则返回公式本身 '=AVERAGE(B2:B8)'

append函数

你可以一次添加多行数据，从第一个空行（以下所有空行）开始写入。

# 添加一行
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 函数只能逐行写入。如果我们想逐列写怎么办。 append可以满足要求吗？如果你将上述列表嵌套视为一个矩阵。只需转置矩阵。这可以使用 zip() 函数来实现，但是内部列表变成了一个元组。所有都是可迭代的对象，不影响。

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

解释一下上面的list(zip(*rows))先*rows打散list，相当于填了几个参数，zip从一个list中取出第一个值组合成一个元组，再从list中取出第一个值每个列表。提取第二个值并组合成一个元组，直到提取出最短列表的最后一个值，丢弃较长列表之后的值。也就是说，最终元组的个数是由每个参数的原始值决定的（Iterable object）是由最短的长度决定的。比如随机删除一个值，最短的list长度为2，data2列（垂直看）的所有值都被丢弃。

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

最后， zip 返回 zip 对象，它看不到数据。只需使用列表进行转换。使用 zip 可以很容易地在列中写入数据。

字体

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

上面的代码以粗斜体指定相等的第 24 行，字体颜色为红色。直接使用单元格的字体属性，并将 Font 对象分配给它。

对齐方式

也就是直接使用cell的属性alignment，指定垂直居中和水平居中。除了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

合并和拆分单元格

所谓合并单元格，就是以合并区域左上角的单元格为基础，覆盖其他单元格，使其成为一个大单元格。

相反，拆分单元格并将大单元格的值返回到其原始左上角位置。

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

合并后只可以往左上角写入数据，也就是区间中:左边的坐标。

如果这些待合并的单元格有数据，则只保留左上角的数据，其余的将被丢弃。也就是说，如果合并前左上角没有写入数据，那么合并后的单元格中就没有数据了。

以下是拆分单元格的代码。拆分后，值回到A1位置。

sheet.unmerge_cells('A1:C3')