本文展示了如何使用openpyxl等一系列包创建新工作表或在已有工作表中插入图片的方法。
第一步:导入包
requests包用于下载图片;
openpyxl包用于围绕excel的一系列操作(3.0.9版本,如果出现奇怪报错一直解决不了建议使用这个版本);
PIL(pillow 9.1.0)包用于图像处理;
io包用于沟通不同包对图像的接收和输出。
import requests
import openpyxl
from openpyxl import Workbook
from openpyxl.drawing.image import Image as xlImage
from openpyxl.utils import get_column_letter
from PIL import Image
from io import BytesIO
第二步:加载已有的excel文件
# 加载 Excel 文件
wb = openpyxl.load_workbook(r'path\test.xlsx')
sheet = wb.active#默认为第一个sheet
# 调整行高和列宽
for row in range(3,sheet.max_row+1):
sheet.row_dimensions[row].height = 50
sheet.column_dimensions['F'].width = 8
第三步:下载图像,插入excel并保存
def download_image(url):
response = requests.get(url)
if response.status_code == 200: # 检查响应状态码
image = Image.open(BytesIO(response.content)) # 创建图像对象
if image.mode != 'RGB':
image = image.convert('RGB')#统一图像类型
return image.resize((40, 50)) # 调整图像大小
# 读取链接并下载图片插入到对应位置
for row in range(3, sheet.max_row + 1): # 假设图片链接在第3行开始,第F列是链接(对应下一行的column=6)
link = sheet.cell(row=row, column=6).value # 获取链接单元格的值
# print('processing no.%s image'%(row-2),end = ';',flush = True)
if link: # 如果链接不为空
# 发送HTTP请求下载图片
max_retry = 3 # 最大重试次数
try_count = 0 # 初始化重试次数计数器
while try_count < max_retry:
try:
# 尝试下载图像
resized_image = download_image(link)
break # 如果下载成功,跳出循环
except OSError as e:
try_count += 1 # 重试次数加1
if try_count == max_retry:
# 如果达到最大重试次数仍然失败,则记录错误信息
with open("图片下载错误.txt", "a") as error_file:
error_file.write(f"Failed after {max_retry} attempts: {link}\n")
else:
# 如果尝试次数未达到最大重试次数,打印错误信息并继续尝试下载
print(f"下载图片失败:{e}. 正在尝试重新下载...")
# 将调整后的图像插入到工作表中
img_bytes = BytesIO()
resized_image.save(img_bytes, format='PNG') # 将图片保存到内存中
img = xlImage(img_bytes)
sheet.add_image(img, f'{get_column_letter(6)}{row}') # 插入图片到指定位置
wb.save(r'path\result.xlsx')#必要
wb.close() #必要
针对读取已有的excel,经过一系列处理后,一定要进行wb.save() 和wb.close()操作,否则无法顺利保存;而对于要创建的excel文件则仅有wb.close()是必要的:
import pandas as pd
#创建新工作表
writer = pd.ExcelWriter(r'path',engine = 'openpyxl')
df_all.to_excel(writer,sheet_name='Sheet1') # 将DataFrame数据导入writer
worksheet = writer.sheets['Sheet1']
worksheet.delete_rows(3) #结果中列名和值之间会有一空行,需要删除
相同的处理操作......
writer.close() # 保存表格文件