#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:]