Как получить доступ к определенной строке в DataFrame Pandas после операции группировки и агрегирования

Question

Я хочу получить доступ к строке в Pandas DataFrame после групповой операции и операции агрегирования.

import pandas as pd
df = pd.DataFrame({'A' : ['foo', 'bar', 'foo', 'bar',
                           'foo', 'bar'],
                    'B' : [1, 2, 3, 4, 5, 6],
                    'C' : [2.0, 5., 8., 1., 2., 9.]})

grouped = df.groupby('A').agg({"C":{"size","mean"}})

Теперь я хочу получить доступ к строке, в которой значение A равно "foo".Когда я попытался использовать grouped[grouped["A"]=="foo"], я получил сообщение об ошибке: KeyError: 'A'

В частности, я хочу размер для "foo".

При поиске в Интернете я увидел несколько сообщенийсвязанные с мультииндексом.Но я не мог заставить его работать.

Это может показаться тривиальным вопросом.Я новичок в Пандах, и мне немного трудно это понять.

Answer 1

Вы должны использовать loc:

grouped.loc['foo']
Out[1]: 
C  size    3.0
   mean    4.0
Name: foo, dtype: float64