создание нескольких данных, которые содержат подмножества исходной части данных pandas

Question

У меня есть dataframe (называемый df) и вы хотите разбить его на несколько фреймов данных на основе значений в одном из столбцов.

Я думаю, что синтаксис будет:

month = df.month.unique().tolist() 
for item in month: 
    [item] = df[df[month]==[item]] 

Answer 1

Объект DataFrameGroupBy итератор, который возвращает вложенные DataFrames:

for month, subdf in df.groupby(['month']): 
    ... 

---

Если вместо этого вы хотите, случайный доступ к суб- DataFrames, указанные в месяце , вы можете изменить столбец month в индекс:

df = df.set_index(['month']) 

, а затем вы можете выбрать строки по месяцам с:

df.loc[month] 

---

Например,

In [4]: df = pd.DataFrame({'month': ['Jan','Jan','Feb'], 'val':[1,2,3]})   
In [6]: df = df.set_index(['month'])  

Учитывая это DataFrame:

In [7]: df 
Out[7]: 
     val 
month  
Jan  1 
Jan  2 
Feb  3 

Это выбирает строки где месяц (index) is 'Jan':

In [8]: df.loc['Jan'] 
Out[8]: 
     val 
month  
Jan  1 
Jan  2