Как получить доступ к колонке после панд .groupby

Question

У меня есть фрейм данных, в котором я использовал .groupby () вместе с функцией .agg ().

movieProperties = combined_df.groupby(['movieId', 'title', 'genres']).agg({'rating': ['count', 'mean']})

Это код для создания нового фрейма данных. Однако я больше не могу получить доступ к столбцам таким же образом. Если я пытаюсь movieProperties['genres'], я всегда получаю KeyError. Как я могу снова получить доступ к столбцам в этом новом фрейме данных?

Answer 1

после того, как вы сгруппировались, столбцы, по которым вы группировались, теперь называются index:

movieProperties = pd.DataFrame({"movie": ["x", "x", "y"], "title":["tx", "tx", "ty"], "rating": [3, 4, 3]}).groupby(["movie", "title"]).agg({"rating":["count", "mean"]})
movieProperties.index.values
Out[13]: array([('x', 'tx'), ('y', 'ty')], dtype=object)

, если вас это не устраивает, сбросьте их обратно на обычные столбцы:

movieProperties.reset_index()
Out[16]: 
  movie title rating     
               count mean
0     x    tx      2  3.5
1     y    ty      1  3.0

, а затем

movieProperties.reset_index()["movie"]
Out[17]: 
0    x
1    y