WebDriver提供了截图函数get_screenshot_as_file(),来截取当前窗口
from selenium import webdriver
import time
# 截图练习,使用get_screenshot_as_file()方法
driver=webdriver.Chrome()
driver.get("http://www.baidu.com")
driver.find_element_by_id("kw").send_keys("selenium")
driver.find_element_by_id("su").click()
time.sleep(10)
# 截取当前窗口,并指定保存位置
driver.get_screenshot_as_file("D:\\baidu_img.png")
driver.quit()
截图效果如下: