selenium 自带的截图方法有两个。
方法一： save_screenshot()
方法二： get_screenshot_as_file()
用法一样，都是截取浏览器当前窗口里的内容。
from selenium import webdriver
driver = webdriver.Chrome()
driver.get("https://www.9958o.com")
# 截图方式一
driver.save_screenshot("test1.jpg")
# 截图方式二
driver.get_screenshot_as_file("test2.jpg")