pandas.core.groupby.DataFrameGroupBy.head

DataFrameGroupBy.head(n=5)

返回每个组的前 n 行。

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

参数:

nint

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

返回:

系列或数据框

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

参见

Series.groupby

对一个Series应用一个groupby函数。

DataFrame.groupby

对DataFrame的每一行或每一列应用一个groupby函数。

例子

>>> 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