python スクレイピングした複数画像データをエクセルに場所を指定して挿入したい

Question

実現したいこと

スクレイピングした複数画像データをエクセルに場所を指定して挿入したい

実施したこと

from PIL import Image
from selenium import webdriver
import  io
from urllib import request
import time
from datetime import date as dt
import pandas as pd
import os


browser = webdriver.Chrome()
browser.get('https://xxx.jp/')
time.sleep(60)

#画像保存は済み
for i in range(6):
    for elem in browser.find_elements_by_xpath('/html/body/main/div/div[1]/div[1]/div/div[1]/ul/li[{}]/a/img'.format(i)):
        try:
                imgs = elem.get_attribute('src')
                image = io.BytesIO(request.urlopen(imgs).read())
                img = Image.open(image)
                img = img.resize((560, 250))
                img.save(r'C:\Users/xxx{}.jpg'.format(i)
        except OSError:
            pass

browser.get('https://yyy.jp/')
time.sleep(60)

for i in range(6):
    for elem in browser.find_elements_by_xpath('/html/body/main/div/div[1]/div[1]/div/div[1]/ul/li[{}]/a/img'.format(i)):
        try:
                imgs = elem.get_attribute('src')
                image = io.BytesIO(request.urlopen(imgs).read())
                img = Image.open(image)
                img = img.resize((560, 250))
                img.save(r'C:\Users/yyy{}.jpg'.format(i)
        except OSError:
            pass

試したこと

import openpyxl
import cv2
wb = openpyxl.Workbook()
ws = wb.worksheets[0]

for i in range(1,5):
    try:
        img = openpyxl.drawing.image.Image(r'C:\Users\xxx{}.jpg'.format(i))
        ws.add_image( img, 'A{}'.format(i))
        wb.save('11.xlsx')
    except FileNotFoundError:
                pass

→複数画像があるのでfor文で記載しておりますが、重なって挿入されてしまいます。

またファイルごと（xxx , yyy）にシートを分け、画像を縦に保存していく方法が分かりません。ご教授いただけるとありがたいです。

何卒宜しくお願い致します。

Answer 1

画像を保存することはできたとの事なので、Excelに画像を貼り付ける方法を回答と致します。
openpyxlを使用するため、「pip install openpyxl」、画像を使用するため「pip install Pillow」を行ったあとに下記のコードで貼り付けが可能です。

import openpyxl

wb = openpyxl.Workbook()
ws = wb.worksheets[0]
img = openpyxl.drawing.image.Image('sample.png')
ws.add_image( img, 'A1' )
wb.save('result.xlsx')

※シートを追加する場合には「wb.create_sheet()」で可能です。

openpyxlのドキュメント↓
https://openpyxl.readthedocs.io/en/stable/

---

追記への回答

ファイル毎に分けるのはファイル名を正規表現などで切り分けてシートを選択、
画像を縦に保存するには単純にセルをずらせばいいのではないでしょうか？

以下のような感じでしょうか。（動かしていないので、動かなかったら適時修正してください。）

import openpyxl
import os
import re

path = 'd:/work/20190807'
output_path = '11.xlsx'
r = re.compile(r'xxx.*') # ファイル名にマッチする正規表現

wb = openpyxl.Workbook()
wb.create_sheet()

files = os.listdir(path)
for i, f in enumerate(files):
    if r.match(f):
        ws = wb.worksheets[0]
    else:
        ws = wb.worksheets[1]
    img = openpyxl.drawing.image.Image(os.path.join(path, f))
    row_number = 1 if i == 0 else i * 100
    ws.add_image( img, 'A{}'.format(row_number))

wb.save(output_path)

尚、画像をきっちり縦に並べたいのであれば、画像の縦のサイズを調べてセルのサイズで割るなどが必要でしょうが、それは本質問の内容とはだいぶずれるので割愛します。

Comments

[退会済みユーザー]

yamap様
ご回答ありがとうございます。

import openpyxl
import cv2
wb = openpyxl.Workbook()
ws = wb.worksheets[0]

for i in range(1,5):
try:
img = openpyxl.drawing.image.Image(r'C:\Users\xxx{}.jpg'.format(i))
ws.add_image( img, 'A{}'.format(i))
wb.save('11.xlsx')
except FileNotFoundError:
pass

→複数画像があるのでfor文で記載しておりますが、重なって挿入されてしまいます。

またファイルごと（xxx , yyy）にシートを分け、画像を縦に保存していく方法が分かりません。ご教授いただけるとありがたいです。