Я пишу код для сохранения данных, но он работает неправильно - PullRequest
Новая
       15

Я пишу код для сохранения данных, но он работает неправильно

1 голос
/ 26 мая 2019

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

for value in set(df['mode']):
    df[df['mode'] == value].to_csv('df_beach_'+str(value)+'.csv', index = False)

Я понимаю, что в итоге должно получиться что-то вроде этого: 

mode
beach 35.213412
boat 41.2341
charter 75.213412
pier 30.02384
name: price, dtype: float64

1 Ответ

1 голос
/ 26 мая 2019

Создайте список из df['mode'] только уникальных значений: 

l = list(set(df['mode'].tolist()))

Затем запустите: 

for value in l:
    df.loc[df['mode'] == value].to_csv('df_beach_'+str(value)+'.csv', index = False)

или 

for value in l:
    df1 = df.loc[df['mode']==value]
    df1.to_csv('df_beach_'+str(value)+'.csv', index = False)

Должны дать необходимые CSV,

...