我想把下面的内容写入csv,应该加些什么代码from selenium import webdriver from selenium.webdriver.common.by import By import time import csv driver = webdriver.Chrome() driver.get('https://movie.douban.com/tv/') #让窗口最大化 driver.maximize_window() time.sleep(2) #点击排序、高分优先:需要分开定义 orderxpath = '//*[@id="app"]/div/div[1]/div/div[1]/div[6]/div' orders = driver.find_element(By.XPATH,orderxpath) orders.click() time.sleep(1) highscorexpath = '//*[@id="app"]/div/div[1]/div/div[1]/div[6]/div[2]/div/ul/li[4]/span' highscore = driver.find_element(By.XPATH,highscorexpath) highscore.click() time.sleep(1) #点击加载更多 morexpath = '//*[@id="app"]/div/div[2]/div/button' more = driver.find_element(By.XPATH,morexpath) for i in range(4): more.click() time.sleep(4) driver.execute_script('window.scrollBy(0,2000)') #爬取电视剧+评分 info = driver.find_elements(By.CLASS_NAME,'drc-subject-info-title-text') score = driver.find_elements(By.CLASS_NAME,'drc-rating-num') j=0 for i ,s in zip(info, score): print(j, i.text, s.text) j+=1
时间: 2024-03-28 16:37:19 浏览: 50
您可以使用以下代码将电视剧和评分数据写入到一个名为 "douban_tv.csv" 的 CSV 文件中:
```python
import csv
from selenium import webdriver
from selenium.webdriver.common.by import By
import time
# 创建 Chrome 浏览器实例
driver = webdriver.Chrome()
# 访问网页并进行操作
driver.get('https://movie.douban.com/tv/')
driver.maximize_window()
time.sleep(2)
# 点击排序、高分优先
orderxpath = '//*[@id="app"]/div/div[1]/div/div[1]/div[6]/div'
orders = driver.find_element(By.XPATH, orderxpath)
orders.click()
time.sleep(1)
highscorexpath = '//*[@id="app"]/div/div[1]/div/div[1]/div[6]/div[2]/div/ul/li[4]/span'
highscore = driver.find_element(By.XPATH, highscorexpath)
highscore.click()
time.sleep(1)
# 点击加载更多
morexpath = '//*[@id="app"]/div/div[2]/div/button'
more = driver.find_element(By.XPATH, morexpath)
for i in range(4):
more.click()
time.sleep(4)
driver.execute_script('window.scrollBy(0,2000)')
# 爬取电视剧和评分
info = driver.find_elements(By.CLASS_NAME, 'drc-subject-info-title-text')
score = driver.find_elements(By.CLASS_NAME, 'drc-rating-num')
data = [[i+1, i.text, s.text] for i, s in enumerate(zip(info, score))] # 将数据整理成列表
# 将数据写入 CSV 文件
with open('douban_tv.csv', mode='w', newline='', encoding='utf-8') as file:
writer = csv.writer(file)
writer.writerow(['序号', '电视剧', '评分']) # 写入 CSV 文件的表头
writer.writerows(data) # 将整理好的数据写入 CSV 文件中
# 关闭浏览器实例
driver.quit()
```
在这个示例代码中,我们首先使用 Selenium 创建了一个 Chrome 浏览器实例,并访问了一个网页。然后,我们进行了一系列操作,包括:点击排序、高分优先;点击加载更多;爬取电视剧和评分。接着,我们将爬取到的数据整理成了一个二维列表 `data`,并使用 Python 的 `csv` 模块将数据写入到 CSV 文件中。最后,我们关闭了 Chrome 浏览器实例。请注意,这里的代码仅供参考,您需要根据实际情况进行调整。
阅读全文