Панды рассчитывают количество месяцев - PullRequest
0 голосов
/ 04 октября 2018

как считать элементы в каждом месяце с заполнением?

data = [
        {"event_date": "2018-08-10", "tags": ["tv", "radio"]},
        {"event_date": "2018-08-11", "tags": ["tv", "radio"]},
        {"event_date": "2018-09-10", "tags": ["tv"]},
        {"event_date": "2018-11-10", "tags": ["tv", "wifi"]}
    ] 
df = pd.DataFrame(data)
df.groupby([(df['event_date']).dt.month, df['tags']]).count()

что я ожидаю:

month tv radio wifi
8 2 2 0
9 1 0 0
10 0 0 0
11 1 0 1

Ответы [ 2 ]

0 голосов
/ 04 октября 2018

Метод, аналогичный @ Wen , но создание нового DataFrame для избежания apply:

s = (pd.DataFrame(df.tags.values.tolist(),
        index=df.event_date.dt.month).stack().reset_index(1, drop=True))

Теперь используется pd.get_dummies:

(pd.get_dummies(s).sum(level=0)
    .reindex(np.arange(s.index.min(),s.index.max()+1), fill_value=0))

            radio  tv  wifi
event_date
8               2   2     0
9               0   1     0
10              0   0     0
11              0   1     1
0 голосов
/ 04 октября 2018

Так что это unnest , get_dummies и reindex смешанный вопрос

df.event_date=pd.to_datetime(df.event_date).dt.month

l=list(range(df.event_date.min(),df.event_date.max()+1))
df.set_index('event_date').tags.apply(pd.Series).stack().\
  str.get_dummies().sum(level=0).\
   reindex(l,fill_value=0)
Out[834]: 
            radio  tv  wifi
event_date                 
8               2   2     0
9               0   1     0
10              0   0     0
11              0   1     1
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...