openpyxl:Python的Excel操作库
openpyxl是Python下的Excel库,它能够很容易的对Excel数据进行读取、写入以及样式的设置,能够帮助我们实现大量的、重复的Excel操作,提高我们的办公效率,实现Excel办公自动化。
- 安装方法:
pip install openpyxl
- 中文文档:https://www.osgeo.cn/openpyxl/index.html#usage-examples
- 工作簿、工作表、单元格之间的关系:
- 一个工作簿(workbook)由多个工作表(worksheet)组成;
- 一个工作表有多个单元格(cell)组成;
- 通过行(row)和列(column)可以定位到单元格。
新建并写入文件
- Workbook():新建excel文件,新建文件时默认有一个名为Sheet工作表
# coding=utf-8
from openpyxl import Workbook
wb = Workbook() # 新建工作簿
ws = wb.active # 获取工作表
ws.append(['姓名', '学号', '年龄']) # 追加一行数据
ws.append(['张三', "1101", 17]) # 追加一行数据
ws.append(['李四', "1102", 18]) # 追加一行数据
wb.save(r'测试1.xlsx') # 保存到指定路径,保存的文件必须不能处于打开状态,因为文件打开后文件只读
打开并读取文件
- load_workbook(path):加载指定路径的excel文件
# coding=utf-8
from openpyxl import load_workbook
wb = load_workbook(r'测试1.xlsx') # 获取已存在的工作簿
ws = wb.active # 获取工作表
for row in ws.values: # 输出所有数据
print(row)
工作簿对象
- wb.active :获取第一张工作表对象
- wb[sheet_name] :获取指定名称的工作表对象
- wb.sheetnames :获取所有工作表名称
- wb.worksheets:获取所有工作表对象,wb.worksheets[0]可以根据索引获取工作表,0代表第一个
- wb.create_sheet(sheet_name,index=“end”):创建并返回一个工作表对象,默认位置最后,0代表第一个
- wb.copy_worksheet(sheet):在当前工作簿复制指定的工作表并返回复制后的工作表对象
- wb.remove(sheet):删除指定的工作表
- ws.save(path):保存到指定路径path的Excel文件中,若文件不存在会新建,若文件存在会覆盖
# coding=utf-8
from openpyxl import load_workbook
wb = load_workbook(r"测试1.xlsx")
"""获取工作表"""
active_sheet = wb.active # 获取第一个工作表
print(active_sheet) # 输出工作表:<Worksheet "Sheet">
by_name_sheet = wb["Sheet"] # 根据工作表名称获取工作表
by_index_sheet = wb.worksheets[0] # 根据工作表索引获取工作表
"""获取所有工作表"""
print("获取所有",wb.sheetnames)
"""新建工作表"""
New_Sheet = wb.create_sheet("New") # 在最后新建工作表
First_Sheet = wb.create_sheet("First",index=0) # 在开头新建工作表
print("新建后",wb.sheetnames)
"""复制工作表"""
Copy_Sheet = wb.copy_worksheet(active_sheet) # 复制第一个工作表
Copy_Sheet.title = "Copy"
print("复制后",wb.sheetnames)
"""删除工作表"""
wb.remove(First_Sheet) # 根据指定的工作表对象删除工作表
wb.remove(New_Sheet)
print("删除后",wb.sheetnames)
wb.save(r"测试2.xlsx")
工作表对象
- ws.title:获取或设置工作表名
- ws.max_row:工作表最大行数
- ws.max_column:工作表最大列数
- ws.append(list):表格末尾追加数据
- ws.merge_cells(‘A2:D2’):合并单元格
- ws.unmerge_cells(‘A2:D2’):解除合并单元格。
# coding=utf-8
from openpyxl import load_workbook
wb = load_workbook(r'测试1.xlsx') # 获取已存在的工作簿
ws = wb.active
print("工作表名",ws.title)
ws.title = "学生信息表"
print("修改后工作表名",ws.title)
print("最大行数",ws.max_row)
print("最大列数",ws.max_column)
ws.append(["王五","1103",17])
print("最大行数",ws.max_row)
wb.save(r"测试3.xlsx")
单元格读取
- ws[‘A1’]:根据坐标获取单个单元格对象
- ws.cell(row, column, value=None):根据行列获取单个单元格对象
- ws[1]:获取第一行所有单元格对象,ws[“1”]也可
- ws[“A”]:获取第A列所有单元格对象
- ws[“A”:“B”]:获取A到B列所有单元格对象,ws[“A:B”]也可
- ws[1:2]:获取1到2行所有单元格对象,ws[“1:2”]也可
- ws[“A1”:“B2”]:获取A1到B2范围所有单元格对象,ws[“A1:B2”]也可。
# coding=utf-8
from openpyxl import load_workbook
wb = load_workbook(r'测试1.xlsx')
ws = wb.active
A1 = ws["A1"] # 根据坐标获取单个单元格
print("第一行第一列",ws.cell(1,1)) # 根据行列获取单个单元格
print("第一行",ws[1])
print("第A列",ws["A"])
print("A到B列",ws["A":"B"])
print("1到2行",ws["1":"2"])
print("A1到B2范围",ws["A1":"B2"])
- ws.values:获取所有单元格数据的可迭代对象,可以通过for循环迭代或通过list(ws.values)转换为数据列表
# coding=utf-8
from openpyxl import load_workbook
wb = load_workbook(r'测试1.xlsx') # 获取已存在的工作簿
ws = wb.active # 获取工作表
for row in ws.values: # for循环迭代
print(row)
print(list(ws.values)) # 转换为数据列表
- ws.rows:获取所有数据以行的格式组成的可迭代对象
- ws.columns:获取所有数据以列的格式组成的可迭代对象
# coding=utf-8
from openpyxl import load_workbook
wb = load_workbook(r'测试1.xlsx')
ws = wb.active
for row in ws.rows: # 以行的形式迭代
print(row)
print("-"*55)
for col in ws.columns: # 以列的形式迭代
print(col)
- ws.iter_rows(min_row=None, max_row=None, min_col=None, max_col=None):获取指定边界范围并以行的格式组成的可迭代对象,默认所有行
- ws.iter_cols(min_col=None, max_col=None, min_row=None, max_row=None): 获取指定边界范围并以列的格式组成的可迭代对象,默认所有列
# coding=utf-8
from openpyxl import load_workbook
wb = load_workbook(r'测试1.xlsx')
ws = wb.active
for row in ws.iter_rows(max_row=2,max_col=2): # 指定边界范围并以行的形式可迭代
print(row)
print("-"*35)
for column in ws.iter_cols(max_row=2,max_col=2): # 指定边界范围并以行的形式可迭代
print(column)
单元格对象
- cell.value :获取或设置值
- cell.column : 数字列标
- cell.column_letter : 字母列标
- cell.row : 行号
- cell.coordinate : 坐标,例如’A1’
- cell.data_type : 数据类型, ’s‘ = string字符串,‘n’ = number数值,会根据单元格值自动判断
- cell.number_format :单元格格式,默认”General“常规,详见excel自定义数据类型
# coding=utf-8
from openpyxl import Workbook
wb = Workbook() # 新建工作簿
ws = wb.active
"""获取与设置单元格值的两种方式"""
cell1 = ws.cell(1,1) # 先获取第一行第一列的单元格对象
cell1.value = 18 # 再设置单元格对象的值
print("值",cell1.value)
print("数字列标",cell1.column)
print("字母列标",cell1.column_letter)
print("行号",cell1.row)
print("坐标",cell1.coordinate)
cell2 = ws.cell(2,1,17) # 直接在获取单元格的时候设置值
"""使用公式和不适用公式"""
cell3 = ws.cell(3,1,"=A1+A2") # 直接输入公式具有计算功能
cell4 = ws.cell(4,1,"=A1+A2")
cell4.data_type = 's' # 指定单元格数据类型为文本可以避免公式被计算
"""设置格式和不设置格式"""
cell5 = ws.cell(5,1,3.1415) # 默认常规格式
cell6 = ws.cell(6,1,3.1415)
cell6.number_format = "0.00" # 设置格式为保留两位小数
wb.save(r'测试4.xlsx') # 保存到指定路径
单元格样式
- cell.font :获取或设置单元格Font对象 (字体名称,字体大小,是否加粗,字体颜色等)
- cell.border : 获取或设置单元格边框
- cell.alignment : 获取或设置单元格水平/垂直对齐方式
- cell.fill:获取或设置单元格填充颜色
from openpyxl import Workbook
from openpyxl.styles import Font, Border, Side, Alignment,PatternFill
from copy import copy
wb = Workbook()
ws = wb.active
"""获取单元格并设置单元格值为 姓名 """
cell = ws.cell(1,1,"姓名")
"""设置单元格文字样式"""
cell.font = Font(bold=True, # 加粗
italic=True, # 倾斜
name="楷体", # 字体
size=13, # 文字大小
color="FF0000" # 字体颜色为红色
)
"""复制单元格样式"""
cell2 = ws.cell(1,2,"学号")
cell2.font = copy(cell.font)
"""设置单元格边框为黑色边框"""
cell.border = Border(bottom=Side(style='thin', color='000000'),
right=Side(style='thin', color='000000'),
left=Side(style='thin', color='000000'),
top=Side(style='thin', color='000000'))
"""设置单元格对齐方式为水平垂直居中"""
cell.alignment = Alignment(horizontal='center',vertical='center')
"""设置单元格底纹颜色为黄色"""
cell.fill = PatternFill(fill_type='solid', start_color='FFFF00')
"""
白色:FFFFFF,黑色:000000,红色:FF0000,黄色:FFFF00
绿色:00FF00,蓝色:0000FF,橙色:FF9900,灰色:C0C0C0
常见颜色代码表:https://www.osgeo.cn/openpyxl/styles.html#indexed-colours
"""
wb.save(r"测试5.xlsx")
列宽与行高
- ws.row_dimensions[行号]:获取行对象(非行数据,包括行的相关属性、行高等)
- ws.column_dimensions[字母列标]:获取列对象(非行数据,包括行的相关属性、列宽等)
- get_column_letter(index):根据列的索引返回字母
- column_index_from_string(string):根据字母返回列的索引
- row.height:获取或设置行高
- column.width:获取或设置列宽
from openpyxl import Workbook
from openpyxl.utils import get_column_letter,column_index_from_string
wb = Workbook()
ws = wb.active
"""行"""
row = ws.row_dimensions[1] # 获取第一行行对象
print("行号",row.index)
row.height = 20 # 设置行高
print("行高",row.height)
"""列"""
column = ws.column_dimensions["A"] # 根据字母列标获取第一列列对象
column = ws.column_dimensions[get_column_letter(1)] # 根据数字列标获取第一列列对象
print("字母列标",column.index)
print("数字列标",column_index_from_string(column.index))
column.width = 15 # 设置列宽
print("列宽",column.width)
wb.save(r'测试6.xlsx')
如何根据输入内容计算其在excel的列宽是多少?
- 利用GBK编码方式,非汉字字符占1个长度,汉字字符占2个长度
from openpyxl import Workbook
from openpyxl.utils import get_column_letter,column_index_from_string
wb = Workbook()
ws = wb.active
column = ws.column_dimensions[get_column_letter(1)] # 根据数字列标获取第一列列对象
value = "我爱中国ILoveChain" # 4*2+10*1+1=19
column.width = len(str(value).encode("GBK"))+1 # 根据内容设置列宽,+1既可以补充误差又可以让两边留有一定的空白,美观
print("列宽",column.width) # 输出:19
ws.cell(1,1,value)
wb.save(r'测试6.xlsx')
插入和删除行和列
插入和删除行、列均使用数字指定
- ws.insert_rows(row_index,amount=1):在第row_index行上方插入amount列,默认插入1列
- ws.insert_cols(col_index,amount=1):在第col_index列左侧插入amount列,默认插入1列
- ws.delete_rows(row_index,amount=1):从row_index行开始向下删除amount行,默认删除1行
- ws.delete_cols(col_index,amount=1):从col_index列开始向右删除amount行,默认删除1列
from openpyxl import Workbook,load_workbook
wb = load_workbook("测试1.xlsx")
ws = wb.active
ws.insert_rows(1,2) # 在第一行前插入两行
delete_col_index = [1,3] # 删除1、3两列
"""为避免删除多列时前面列对后面列产生影响,采取从后面列往前面列删的策略,行同理"""
delete_col_index.sort(reverse=True) # 从大到小排序
for col_index in delete_col_index:
ws.delete_cols(col_index)
wb.save(r'测试7.xlsx')
综合写入实践
写入后的效果如下:
# coding=utf-8
from openpyxl import Workbook
from openpyxl.styles import Font, Border, Side, Alignment, PatternFill
wb = Workbook()
ws = wb.active
"""设置全局样式"""
border = Border(bottom=Side(style='thin', color='000000'),
right=Side(style='thin', color='000000'),
left=Side(style='thin', color='000000'),
top=Side(style='thin', color='000000'))
alignment = Alignment(horizontal='center', vertical='center')
row_index = 1 # 写入的行索引,每写入一行后+1
"""写入标题"""
title = ['姓名', '学号', '分数']
for index,item in enumerate(title):
cell = ws.cell(row_index,index+1,item)
cell.border = border
cell.alignment = alignment
cell.font = Font(bold=True)
row_index += 1
data = [['张三', "1101", 17],['李四', "3412", 18],['王五', "1103", 16]]
"""写入正文"""
for row in data:
for index,item in enumerate(row):
cell = ws.cell(row_index, index + 1, item)
cell.border = border
cell.alignment = alignment
row_index += 1
"""写入结果"""
result = ["", "合计", 17+18+16]
for index,item in enumerate(result):
cell = ws.cell(row_index,index+1,item)
cell.border = border
cell.alignment = alignment
cell.fill = PatternFill(fill_type='solid', start_color="FFFF00")
wb.save(r"学生信息表.xlsx")
合并表格
# coding=utf-8
from openpyxl import Workbook,load_workbook
import os
dir_path = "学生名单" # 要合并文件的文件夹地址
"""读取文件夹下的所有excel文件"""
files = []
for file in os.listdir(dir_path): # 获取当前目录下的所有文件
files.append(os.path.join(dir_path,file)) # 获取文件夹+文件名的完整路径
"""以第一个文件为基本表"""
merge_excel = load_workbook(files[0])
merge_sheet = merge_excel.active
"""遍历剩余文件,追加到基本表"""
for file in files[1:]:
wb = load_workbook(file)
ws = wb.active
for row in list(ws.values)[1:]: # 从第二行开始读取每一行并追加到基本表
merge_sheet.append(row)
merge_excel.save("高一学生汇总.xlsx")
拆分表格
# coding=utf-8
from openpyxl import Workbook,load_workbook
import os
file_path = "高一学生汇总.xlsx" # 要拆分的文件地址
split_dir = "拆分结果" # 拆分文件后保存的文件夹
group_item = "班级" # 拆分的依据字段
"""打开拆分的excel文件并读取标题"""
wb = load_workbook(file_path)
ws = wb.active
title = []
for cell in ws[1]:
title.append(cell.value)
"""开始分组,分组结果保存到字典,键为班级名,值为班级学生列表"""
group_result = {} # 存储分组结果
group_index = title.index(group_item) # 获取拆分依据字段的索引
for row in list(ws.values)[1:]:
class_name = row[group_index] # 获取分组依据数据,即班级名
if class_name in group_result: # 如果分组存在就追加,不存在就新建
group_result[class_name].append(row)
else:
group_result[class_name] = [row]
"""创建输出文件夹"""
if not os.path.exists(split_dir): # 如果不存在文件夹就新建
os.mkdir(split_dir)
os.chdir(split_dir) # 进入拆分文件夹
"""打印并输出分组后的数据"""
for class_name,students in group_result.items():
new_wb = Workbook() # 新建excel
new_ws = new_wb.active
new_ws.append(title) # 追加标题
for student in students:
new_ws.append(student) # 讲分组数组追加到新excel中
new_wb.save("{}.xlsx".format(class_name))
作业提交情况检测
# encoding: utf-8
import os
from openpyxl import Workbook, load_workbook
excel_path = r"学生名单/高一1班.xlsx" # excel文件路径
job_path = r"作业" # 作业文件夹路径
"""获取姓名列表"""
wb = load_workbook(excel_path)
ws = wb.active
names = []
for cell in ws["C"][1:]: # 获取第C列第2行开始的数据
names.append(cell.value)
"""获取作业列表"""
os.chdir(job_path) # 切换到作业目录
files = [] # 获取文件列表
for file in os.listdir():
files.append(os.path.splitext(file)[0])
"""作业检测"""
yes,no = [],[]
for name in names: # 逐个姓名判断
if name in files: # 判断姓名是否在文件列表中
yes.append(name) # 如果在,添加到已完成名单
else:
no.append(name) # 否则,添加到未完成名单
print("已完成人数:{},已完成名单:{}".format(len(yes),yes))
print("未完成人数:{},未完成名单:{}".format(len(no),no))