Я пытаюсь перебрать список, выполнить поиск на веб-странице через селен и сохранить результаты в файле df. Как сохранить результаты цикла из каждого элемента списка в df?
from selenium.webdriver.common.keys import Keys
import pandas as pd
import numpy as np
url = 'https://au.finance.yahoo.com/australia/'
driver_path = 'chromedriver.exe'
browser = Chrome(executable_path= driver_path)
loop_search = browser.find_element_by_id('yfin-usr-qry')
search_companies = ['Commonwealth Bank','Rio Tinto','Wesfarmers']
for i in search_companies:
loop_search.send_keys(i)
browser.find_element_by_id('search-button').click()
comp = browser.find_element_by_id('quote-header-info').text
df3 = [comp]```
Still fairly new to Python! Thank you!






Если вы просто запустите свой код и сделаете print(comp)
вы увидите следующую ошибку:
selenium.common.exceptions.StaleElementReferenceException: Message: stale element reference: element is not attached to the page document
(Session info: chrome=99.0.4844.74)
поэтому, прежде чем сохранять его в DF, нам нужно решить это:
это можно исправить, переопределив веб-элемент следующим образом в цикле:
loop_search = wait.until(EC.visibility_of_element_located((By.ID, "yfin-usr-qry")))
Полный код для сохранения в DF:
driver_path = 'chromedriver.exe'
browser = Chrome(executable_path= driver_path)
wait = WebDriverWait(driver, 20)
url = 'https://au.finance.yahoo.com/australia/'
driver.get(url)
search_companies = ['Commonwealth Bank','Rio Tinto','Wesfarmers']
company_details_lst = []
for i in search_companies:
time.sleep(2)
loop_search = wait.until(EC.visibility_of_element_located((By.ID, "yfin-usr-qry")))
loop_search.send_keys(i)
time.sleep(2)
wait.until(EC.element_to_be_clickable((By.ID, "search-button"))).click()
time.sleep(2)
comp = wait.until(EC.element_to_be_clickable((By.ID, "quote-header-info"))).text
company_details_lst.append(comp)
#print(comp)
data = {
'Details': company_details_lst
}
df = pd.DataFrame.from_dict(data)
df.to_csv('output.csv', index = 0
Импорт:
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
после запуска вашего кода вы должны увидеть файл csv в папке вашего проекта с именем output.csv
и внутреннее содержание будет:
Это была проблема синхронизации данных, которую можно легко решить с помощью time. sleep(2), см. выше также измененный код, вы можете увеличить 2 до 5 или 7 в зависимости от успешности выполнения вашего кода.
Спасибо @cruisepandey за исчерпывающий ответ и объяснение! Тем не менее, я хотел также перебрать список search_companies, чтобы заполнить его в файле df. как мне получить информацию о других пунктах списка Rio Tinto и Wesfarmers?