将 Pandas 数据框转换为列表列表答案

【问题标题】：Pandas dataframe into a list of lists将 Pandas 数据框转换为列表列表
【发布时间】：2021-03-22 01:21:47
【问题描述】：

我也在尝试读取一个 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'

【问题讨论】：

标签： python pandas list dataframe csv

【解决方案1】：

你可以直接在你想成为列表列表的列上使用 values.tolist()。

您不需要循环和创建列表。

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

要从列表中排除前 4 条记录，您现在可以从 data_list 中删除它

data_list=data_list[4:]

【讨论】：