如何为Jupyter html输出保存图像快照？

pip install selenium
pip install webdriver_manager

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.chrome.options import Options
from webdriver_manager.chrome import ChromeDriverManager
import time

options = Options()
options.add_argument("--headless")  # 无头模式，不显示浏览器窗口
driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)

driver.get("http://localhost:8888/notebooks/your_notebook.ipynb")

time.sleep(5)  # 等待页面加载完成，根据需要调整等待时间

element = driver.find_element(By.XPATH, "//div[@class='output_html rendered_html']")
element.screenshot("screenshot.png")

driver.quit()