pandas.core.groupby.GroupBy.nth

GroupBy.nth(self, n: Union[int, List[int]], dropna: Union[str, NoneType] = None) → pandas.core.frame.DataFrame [source]

Получение n-й строки из каждой группы, если n — целое число, или подмножества строк, если n — список целых чисел.

Если dropna, то берется n-я ненулевая строка, dropna может быть ‘all’ или ‘any’; это эквивалентно вызову dropna(how=dropna) перед группировкой.

Параметры:	n : int or list of ints единственное значение n для строки или список значений n dropna : None or str, optional применить указанную операцию dropna перед подсчетом, какая строка является n-й строкой. Должно быть None, ‘any’ или ‘all’
Возвращает:	Series или DataFrame N-е значение в каждой группе.

См. также

Series.groupby

DataFrame.groupby

Примеры

>>> df = pd.DataFrame({'A': [1, 1, 2, 1, 2],
...                    'B': [np.nan, 2, 3, 4, 5]}, columns=['A', 'B'])
>>> g = df.groupby('A')
>>> g.nth(0)
     B
A
1  NaN
2  3.0
>>> g.nth(1)
     B
A
1  2.0
2  5.0
>>> g.nth(-1)
     B
A
1  4.0
2  5.0
>>> g.nth([0, 1])
     B
A
1  NaN
1  2.0
2  3.0
2  5.0

Указание dropna позволяет подсчету игнорировать NaN

>>> g.nth(0, dropna='any')
     B
A
1  2.0
2  3.0

NaN обозначают исчерпание группы при использовании dropna

>>> g.nth(3, dropna='any')
    B
A
1 NaN
2 NaN

Указание as_index=False в groupby сохраняет исходный индекс.

>>> df.groupby('A', as_index=False).nth(1)
   A    B
1  1  2.0
4  2  5.0