Кнопка экспорта Selenium CSV в панды датафрейм?

Question

Я использую селен, чтобы нажать на кнопку, которая позволяет экспортировать данные в формате CSV.Однако я хотел бы знать, можно ли вместо загрузки файла поместить данные в DataFrame или в переменную?

Это команда, которую я использую для загрузки файла:

driver.execute_script("window.open('" + export_url + "');")

Спасибо за помощь

Answer 1

На самом деле, вы можете напрямую использовать функцию pandas.read_csv() для загрузки файла и преобразования в Dataframe.

Параметр filepath_or_buffer разрешает внешние URL-адреса. Смотрите пример ниже:

import pandas as pandas

export_url = 'http://samplecsvs.s3.amazonaws.com/Sacramentorealestatetransactions.csv'
df = pandas.read_csv(filepath_or_buffer=export_url, sep=',', lineterminator='\r')
print(df)

Выход:

                              street             city  ...   latitude   longitude
0                       3526 HIGH ST       SACRAMENTO  ...  38.631913 -121.434879
1                        51 OMAHA CT       SACRAMENTO  ...  38.478902 -121.431028
2                     2796 BRANCH ST       SACRAMENTO  ...  38.618305 -121.443839
3                   2805 JANETTE WAY       SACRAMENTO  ...  38.616835 -121.439146
4                    6001 MCMAHON DR       SACRAMENTO  ...  38.519470 -121.435768
5                 5828 PEPPERMILL CT       SACRAMENTO  ...  38.662595 -121.327813