如何將熊貓數據框信息存儲在 csv 文件中

Question

我是抓取和 python 的新手。 我正在嘗試從此 URL 中抓取多個表： https://en.wikipedia.org/wiki/List_of_Game_of_Thrones_episodes 。 我進行了抓取，現在我正在嘗試將數據幀保存到 csv 文件中。 我試過了，但它只存儲頁面中的第一個表。

代碼：

from pandas.io.html import read_html
page = 'https://en.wikipedia.org/wiki/List_of_Game_of_Thrones_episodes'

wikitables = read_html(page, index_col=0, attrs={"class":"wikitable plainrowheaders wikiepisodetable"})

print ("Extracted {num} wikitables".format(num=len(wikitables)))



for line in range(7):
  df= pd.DataFrame(wikitables[line].head())
  df.to_csv('file1.csv')

Answer 1

您需要將數據幀列表重塑為單個數據幀，然后需要將其導出到 csv 文件。

wikitable = wikitables[0]

for i in range(1,len(wikitables)):
  wikitable = wikitable.append(wikitables[i],sort=True)

wikitable.to_csv('wikitable.csv')

Answer 2

你忘了

import pandas as pd

但您不需要它，因為read_html提供了數據幀列表，您不必將其轉換為數據幀。 可以直接寫。

from pandas.io.html import read_html

url = 'https://en.wikipedia.org/wiki/List_of_Game_of_Thrones_episodes'

wikitables = read_html(url, index_col=0, attrs={"class":"wikitable plainrowheaders wikiepisodetable"})

print("Extracted {num} wikitables".format(num=len(wikitables)))

for i, dataframe in enumerate(wikitables):
    dataframe.to_csv('file{}.csv'.format(i))