Pandas dataframe groupby вызывает удаление столбцов

Question

У меня есть фрейм данных pandas, который нужно сгруппировать по текстовому столбцу, чтобы получить сумму дублированных значений вдоль этого столбца. Но когда я запускаю метод groupby, он загадочным образом сбрасывает многие столбцы. Может ли кто-нибудь помочь мне в этом?

Answer 1

Попробуйте проверить столбцы dtypes, сумма будет только для числового значения.

Например, у вас есть df, как показано ниже:

df=pd.DataFrame({'V1':[1,2,3],'V2':['A','B','C'],'KEY':[1,2,2]})



df.dtypes
Out[159]:
KEY     int64
V1      int64
V2     object
dtype: object

Затем вы набираете groupby и делаете sum для всего кадра данных, он вернет только результат числовых столбцов

df.groupby('KEY').sum()

Out[160]: 
     V1
KEY    
1     1
2     5

Если вам нужен строковый тип для объединения, вы можете

df.groupby('KEY',as_index=False).apply(lambda x : x.sum())
Out[164]: 
   KEY  V1  V2
0    1   1   A
1    4   5  BC