Почему мой мультииндексный фрейм данных имеет повторяющиеся значения для индексов? - PullRequest
Новая
       15

Почему мой мультииндексный фрейм данных имеет повторяющиеся значения для индексов?

0 голосов
/ 22 января 2019

У меня есть следующий pd.DataFrame с именем df: 

                   date     cluster_label        value
0   2018-11-14 02:16:22                 0          1.5
1   2018-11-14 02:16:22                 0          7.0
2   2018-11-14 02:16:22                 0          2.5
3   2018-11-14 02:16:22                 1          3.0
4   2018-11-14 02:16:22                 1          0.5
5   2018-11-14 02:16:22                 2          1.0

Я выполняю следующую команду, чтобы преобразовать столбец даты, чтобы в нем были только значения месяца и года, перед настройкой многоуровневого индекса или кадра данных:

self.df['date'] = self.df['date'].dt.to_period('M')
self.df.set_index(['cluster_label', 'date'], inplace=True)

Теперь вывод выглядит примерно так: 

                           value
cluster_label date                                                                  
0              2018-11     1.5
               2018-11     7.0
               2018-11     2.5
1              2018-11     3.0
               2018-11     0.5
2              2018-11     1.0

Но это не так.Я хочу, чтобы на выходе не было повторяющихся индексов для столбца даты.Вывод должен выглядеть следующим образом: 

                           value
cluster_label date                                                                  
0              2018-11     1.5
                           7.0
                           2.5
1              2018-11     3.0
                           0.5
2              2018-11     1.0

Что я делаю не так и как я могу изменить свой код, чтобы получить желаемый вывод?

1 Ответ

0 голосов
/ 22 января 2019

Один из вариантов - добавить cumcount уровень: 

df.set_index(df.groupby(level=[0,1]).cumcount(), append=True)

                         value
cluster_label date            
0             2018-11 0    1.5
                      1    7.0
                      2    2.5
1             2018-11 0    3.0
                      1    0.5
2             2018-11 0    1.0

Где, 

df.set_index(df.groupby(level=[0,1]).cumcount(), append=True).index 
# MultiIndex(levels=[[0, 1, 2], [2018-11], [0, 1, 2]],
#            labels=[[0, 0, 0, 1, 1, 2], [0, 0, 0, 0, 0, 0], [0, 1, 2, 0, 1, 0]],
#            names=['cluster_label', 'date', None])

Другой вариант (я не рекомендую это) - этозамаскируйте эти значения явным образом и сбросьте индекс. 

u = np.where(df.index.duplicated(), '', df.index.get_level_values(1))
df.index = pd.MultiIndex.from_arrays([df.index.get_level_values(0), u])

df
                       value
cluster_label               
0             2018-11    1.5
                         7.0
                         2.5
1             2018-11    3.0
                         0.5
2             2018-11    1.0
...