Pandas dataframe 進入列表列表

Question

我也在嘗試讀取 csv 文件並根據以下原則將其轉換為二維數組：[ [12, 22],[11,21] ] 等。我使用了以下代碼

import pandas as pd
fields = ['month', '8.2']
data = pd.read_csv('demand_with_station_id.csv',   index_col=1 ,usecols = fields)
df = pd.DataFrame(data)

print(df)
supralist = []
for indx, val in df.iterrows():
    lst = [val[indx]['month'], val[indx]['8.2']]
    supralist.append(lst)
print(supralist)

這會導致錯誤。 我不想將前四行讀入列表，我該怎么做？ csv 文件有點不標准，看起來像這樣： csv

我收到以下錯誤：

Traceback (most recent call last):
 
During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 2, in <module>
KeyError: 'Friday'

Answer 1

您可以直接在要成為列表列表的列上使用 values.tolist() 。

你不需要循環和創建列表。

data_list=df[['month', '8.2']].values.tolist()

要從列表中排除前 4 條記錄，您現在可以從 data_list 中刪除它

data_list=data_list[4:]