from selenium import webdriver
driver = webdriver.Chorme()
driver.get("http://www.baidu.com")
截取当前窗口，并指定截图图片的存放位置
发现python3在截图时只支持png的图片格式
driver.get_screenshot_as_file("C:\Users\Public\Pictures\error.png")
driver.quit()
原文：https://www.cnblogs.com/97xiaolai/p/11707661.html