Python (Selenium) -如何将第1页到最后一页的数据保存到同一CSV

from selenium import webdriver
import csv

# 创建一个Chrome浏览器实例
driver = webdriver.Chrome()

# 打开目标网页的第1页
driver.get("目标网页的URL")

# 获取总页数
total_pages = driver.find_element_by_xpath("获取总页数的XPath").text

# 创建一个CSV文件，并写入表头
csv_file = open("data.csv", "w", newline="")
csv_writer = csv.writer(csv_file)
csv_writer.writerow(["数据列1", "数据列2", "数据列3"])  # 根据实际情况修改表头

# 循环遍历每一页
for page in range(1, int(total_pages) + 1):
    # 获取当前页的数据
    data_elements = driver.find_elements_by_xpath("获取数据元素的XPath")

    # 将数据写入CSV文件
    for data_element in data_elements:
        data = [
            data_element.find_element_by_xpath("数据列1的XPath").text,
            data_element.find_element_by_xpath("数据列2的XPath").text,
            data_element.find_element_by_xpath("数据列3的XPath").text,
        ]
        csv_writer.writerow(data)

    # 点击下一页按钮
    next_button = driver.find_element_by_xpath("下一页按钮的XPath")
    next_button.click()

# 关闭CSV文件和浏览器
csv_file.close()
driver.quit()