pandas.core.groupby.DataFrameGroupBy.head

DataFrameGroupBy.head(n=5)

返回每个组的前 n 行。

类似于 .apply(lambda x: x.head(n))，但它返回原始 DataFrame 的行子集，并保留原始索引和顺序（忽略 as_index 标志）。

Parameters:

nint

如果为正数：从每个组的开头包含的条目数。如果为负数：从每个组的末尾排除的条目数。

Returns:

Series 或 DataFrame

由 n 确定的原始 Series 或 DataFrame 的子集。

参见

Series.groupby

将函数 groupby 应用于 Series。

DataFrame.groupby

将函数 groupby 应用于 DataFrame 的每一行或每一列。

Examples

>>> df = pd.DataFrame([[1, 2], [1, 4], [5, 6]],
...                   columns=['A', 'B'])
>>> df.groupby('A').head(1)
   A  B
0  1  2
2  5  6
>>> df.groupby('A').head(-1)
   A  B
0  1  2