Превращает фрейм данных Pandas в список списков

#python #pandas #list #dataframe #csv

#python #pandas #Список #фрейм данных #csv

Вопрос:

Я тоже пытаюсь прочитать CSV-файл и преобразовать его в 2d-массив в соответствии со следующим принципом: [ [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'
 

Ответ №1:

вы можете напрямую использовать значения.tolist() для столбцов, которые вы хотите сделать списком списка.

вам не нужно зацикливаться и создавать список.

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

чтобы исключить первые 4 записи из списка, теперь вы можете удалить его из формы data_list

 data_list=data_list[4:]