Python：openpyxl在excel中读取url并下载、插入图片

原创已于 2024-04-23 13:52:10 修改 · 1.6k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python #excel #开发语言

于 2024-04-23 10:24:19 首次发布

本文详细介绍了如何利用openpyxl、requests和PIL等库在Excel中创建新工作表，加载现有文件，并下载图片插入到指定位置。作者提供了下载图片的函数和处理可能遇到的错误方法。

本文展示了如何使用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() # 保存表格文件