python - Selenium CSV导出按钮到 Pandas 数据框？

我正在使用硒单击允许将数据导出为csv的按钮。但是我想知道是否可以将数据放在DataFrame或变量中而不是下载文件？

那就是我用来下载文件的命令：

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

感谢帮助

最佳答案

实际上，您可以直接使用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