Pandasデータフレームに画像を挿入できますか？

Question

データフレームに画像を挿入

指定URLからスクレイピングした情報を、Pandasでデータフレームに入れています。
そこでスクレイピングした情報と画像も同じ表（別カラム）に挿入したいです。

以下のような感じです。

画像　　　　　　　　|　情報
ーーーーーーーーーーーーーーーーーーーーーーー
0|　画像表示される　|　文字
ーーーーーーーーーーーーーーーーーーーーーーー
1|　画像表示される　|　文字
ーーーーーーーーーーーーーーーーーーーーーーー

URLをリストに

今はスクレイピングした項目をリストに入れています。
URLもリストになっていますが、そこからどうしたら画像として挿入できるか不明です。

コード

from selenium import webdriver
import pandas as pd
browser = webdriver.Chrome()
browser.get('url')

#画像
logos = browser.find_elements_by_class_name('class_name')
image_info = []

for logo in logos:
    logo = logo.find_element_by_tag_name('img')
    img = logo.get_attribute('src')
    image_info.append(img)

df = pd.DataFrame()
df['画像'] = image_info

df

問題点

このままですと、画像のURLリンクがコラムに入りますが、

画像は表示されず、文字として入力されてしまいます。

画像としてデータフレームに挿入することは可能でしょうか？

よろしくお願いします。

Answer 1

このままですと、画像のURLリンクがコラムに入りますが、

画像は表示されず、文字として入力されてしまいます。

別途、画像データをダウンロードする必要があります。requestsを使えば比較的簡単にできます。

画像としてデータフレームに挿入することは可能でしょうか？

「画像として」という要件がよく分かりませんが、バイナリデータそのまま挿入することができます。

以上の動作をするサンプルコードを示します。

from selenium import webdriver
driver = webdriver.Chrome(executable_path=r"～")
url = 'https://www.python.org/'
driver.get(url)

from urllib.parse import urljoin
import requests

srcs,imgs = [],[] # ソースURLと画像データ
elems = driver.find_elements_by_tag_name('img')
for e in elems:
    src = e.get_attribute('src')
    img = requests.get(urljoin(url, src)).content # 画像データをダウンロード
    srcs.append(src)
    imgs.append(img)
driver.quit()

# データフレームを構築し、CSVに保存
import pandas as pd
df = pd.DataFrame({'src':srcs,'img':imgs})
df.to_csv('ret.csv',index=False)

ret.csv

src,img
https://www.python.org/static/img/python-logo.png,"b'\x89PNG\r\n\x1a～以降省略～

ただし画像をデータフレームに入れるくらいなら、別途画像用ディレクトリにダウンロードしておき、データフレームには画像ファイルへのパス情報だけを持たせるほうがよいと思います。
理由については以下ページ参照ください。
データベースに画像を保存するのはありでしょうか？

Comments

[退会済みユーザー]

> can110様
回答ありがとうございます。大変参考になりました。
データフレームにではなく、画像ファイルのパスを入れるようにしました。