Как построить столбцы с месяцами из серии datetime?

Question

Итак, у меня есть эти данные с идентификатором, датой, и если они делали действия в этот день

ID             DATE              DONE  
1           2009-1-3             Yes
1           2009-1-15            Yes
2           2009-9-3             Yes
3           2009-10-7            No
1           2009-12-1            Yes
3           2009-10-8            Yes
2           2009-10-21           Yes

Поэтому я хочу сгруппировать и создать столбцы с месяцами и 1, если они выполняли действия в этом месяце 0 o.w

ID 2009-1 2009-2 ... 2009-9 2009-10 2009-11 2009-12
1     1     0    ...   0         0       0     1
2     0     0    ...   1         1       0     0
3     0     0    ...   0         0       0     0

Спасибо

Answer 1

Еще один способ

Setup

df.DATE = pd.to_datetime(df.DATE)
df.DONE = df.DONE.map({'Yes':1, 'No':0})

---

Solution

Затем создайте вспомогательный столбец int в стиле 200901

df['DATE2']=df.DATE.dt.year.astype(str).add(df.DATE.dt.month.astype(str).str.zfill(2)).astype(int)

и pivot_table + reindex

df.pivot_table(columns='DATE2', index='ID', values='DONE').reindex(np.arange(200901, 200913), axis='columns').fillna(0)


DATE2   200901  200902  200903  200904  200905  200906  200907  200908  200909  200910  200911  200912
ID                                              
1       1.0     0.0     0.0     0.0     0.0     0.0     0.0     0.0     0.0     0.0     0.0     1.0
2       0.0     0.0     0.0     0.0     0.0     0.0     0.0     0.0     1.0     1.0     0.0     0.0
3       0.0     0.0     0.0     0.0     0.0     0.0     0.0     0.0     0.0     0.5     0.0     0.0

Answer 2

Я знаю, что опаздываю, но я думаю, что это решение тоже неплохое:

df['DATE'] = pd.to_datetime(df['DATE']).dt.strftime('%Y-%m')
(df.pivot(columns='DATE', values='DONE') == 'Yes') + 0

Answer 3

Используя Series groupby + all, затем unstack

s=df['DONE'].eq('Yes').\
      groupby([df['ID'],df.DATE.dt.strftime('%Y-%m')]).\
          all().astype(int).unstack(fill_value=0)
s
Out[1026]: 
DATE  2009-01  2009-09  2009-10  2009-12
ID                                      
1           1        0        0        1
2           0        1        1        0
3           0        0        0        0