extract base64 image in jupyter notebook

extract_base64_image_in_notebook.py

import json
import base64

# ipynbファイルを読み込む
with open("notebook.ipynb", "r", encoding="utf-8") as f:
    notebook = json.load(f)

# セルを解析して画像データを抽出
for cell in notebook.get("cells", []):
    if cell.get("cell_type") == "markdown":
        for attachment_name, attachment_data in cell.get("attachments", {}).items():
            # Base64データを取得
            base64_data = attachment_data.get("image/png", "")
            if base64_data:
                # PNGファイルとして保存
                with open(f"{attachment_name}.png", "wb") as img_file:
                    img_file.write(base64.b64decode(base64_data))
                print(f"Saved: {attachment_name}.png")