[Python笔记] 读取EXCEL中的网页链接，打开并截图

大致思路：用Pandas读取EXCEL中的数据，将链接存放入列表中，再用Selenium打开并截图，保存到data文件夹中。

文件结构

EXCEL文件中的内容

代码

from selenium import webdriver
import time
import pandas as pd


df = pd.read_excel("urls.xlsx", header=None)
urls = df[0].tolist()

for i in range(len(urls)):
    driver = webdriver.Chrome()
    driver.get(urls[i])
    time.sleep(2)
    try:
        driver.get_screenshot_as_file("data/{}.png".format(i+1))
    except BaseException as msg:
        print(msg)
    
    driver.quit()

运行效果

截取图片