openpyxl是一个第三方库,可以处理xlsx格式的excel文件。pip install openpyxl安装。
读取excel文件
需要导入相关函数
1
2
3
|
from openpyxl import load_workbook # 默认可读写,若有需要可以指定write_only和read_only为true wb = load_workbook( 'pythontab.xlsx' ) |
默认打开的文件为可读写,若有需要可以指定参数read_only为true。
获取工作表--sheet
1
2
3
4
5
6
7
8
|
# 获得所有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 |
获取单元格
1
2
3
4
5
6
7
|
# 获取某个单元格的值,观察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则是那个单元格的值。另外cell还有一个属性coordinate, 像b4这个单元格返回的是坐标b4。
获得最大行和最大列
1
2
3
|
# 获得最大列和最大行 print (sheet.max_row) print (sheet.max_column) |
获取行和列
sheet.rows为生成器, 里面是每一行的数据,每一行又由一个tuple包裹。
sheet.columns类似,不过里面是每个tuple是每一列的单元格。
1
2
3
4
5
6
7
8
|
# 因为按行,所以返回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对象。
1
2
|
for cell in list (sheet.rows)[ 2 ]: print (cell.value) |
如何获得任意区间的单元格?
可以使用range函数,下面的写法,获得了以a1为左上角,b3为右下角矩形区域的所有单元格。注意range从1开始的,因为在openpyxl中为了和excel中的表达方式一致,并不和编程语言的习惯以0表示第一个值。
1
2
3
4
5
6
7
8
9
10
11
|
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']返回一个tuple,该元组内部还是元组,由每行的单元格构成一个元组。
1
2
3
4
5
6
7
8
9
10
|
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
1
2
3
4
5
|
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
1
2
|
from openpyxl import workbook wb = workbook() |
这样就新建了一个新的工作表(只是还没被保存)。
若要指定只写模式,可以指定参数write_only=true。一般默认的可写可读模式就可以了。
1
2
3
4
5
6
7
8
|
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] |
写入单元格
还可以使用公式
1
2
3
4
|
# 直接给单元格赋值就行 sheet[ 'a1' ] = 'good' # b9处写入平均值 sheet[ 'b9' ] = '=average(b2:b8)' |
但是如果是读取的时候需要加上data_only=true这样读到b9返回的就是数字,如果不加这个参数,返回的将是公式本身'=average(b2:b8)'
append函数
可以一次添加多行数据,从第一行空白行开始(下面都是空白行)写入。
1
2
3
4
5
6
7
8
9
10
11
12
13
|
# 添加一行 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()函数可以实现,不过内部的列表变成了元组就是了。都是可迭代对象,不影响。
1
2
3
4
5
|
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将列表打散,相当于填入了若干个参数,zip从某个列表中提取第1个值组合成一个tuple,再从每个列表中提取第2个值组合成一个tuple,一直到最短列表的最后一个值提取完毕后结束,更长列表的之后的值被舍弃,换句话,最后的元组个数是由原来每个参数(可迭代对象)的最短长度决定的。比如现在随便删掉一个值,最短列表长度为2,data2那一列(竖着看)的值全部被舍弃。
1
2
3
4
5
6
7
8
9
10
11
|
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对象,看不到数据的。使用list转换下就好了。使用zip可以方便实现将数据按列写入。
保存文件
所有的操作结束后,一定记得保存文件。指定路径和文件名,后缀名为xlsx。
1
|
wb.save(r 'd:\example.xlsx' ) |
设置单元格风格--style
先导入需要的类
1
|
from openpyxl.styles import font, colors, alignment |
分别可指定字体相关,颜色,和对齐方式。
字体
1
2
|
bold_itatic_24_font = font(name = '等线' , size = 24 , italic = true, color = colors.red, bold = true) sheet[ 'a1' ].font = bold_itatic_24_font |
上面的代码指定了等线24号加粗斜体,字体颜色红色。直接使用cell的font属性,将font对象赋值给它。
对齐方式
也是直接使用cell的属性aligment,这里指定垂直居中和水平居中。除了center,还可以使用right、left等等参数。
1
2
|
# 设置b1中的数据垂直居中和水平居中 sheet[ 'b1' ].alignment = alignment(horizontal = 'center' , vertical = 'center' ) |
设置行高和列宽
有时候数据太长显示不完,就需要拉长拉高单元格。
1
2
3
4
|
# 第2行行高 sheet.row_dimensions[ 2 ].height = 40 # c列列宽 sheet.column_dimensions[ 'c' ].width = 30 |
合并和拆分单元格
所谓合并单元格,即以合并区域的左上角的那个单元格为基准,覆盖其他单元格使之称为一个大的单元格。
相反,拆分单元格后将这个大单元格的值返回到原来的左上角位置。
1
2
3
|
# 合并单元格, 往左上角写入数据即可 sheet.merge_cells( 'b1:g1' ) # 合并一行中的几个单元格 sheet.merge_cells( 'a1:c3' ) # 合并一个矩形区域中的单元格 |
合并后只可以往左上角写入数据,也就是区间中:左边的坐标。
如果这些要合并的单元格都有数据,只会保留左上角的数据,其他则丢弃。换句话说若合并前不是在左上角写入数据,合并后单元格中不会有数据。
以下是拆分单元格的代码。拆分后,值回到a1位置。
1
|
sheet.unmerge_cells( 'a1:c3' ) |
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://www.pythontab.com/html/2018/pythonhexinbiancheng_0503/1286.html