根据常量列值选择前几行

Question

我希望从列ID值恒定的数据帧中获取位置（前2个）的记录，然后在列值更改时获取下2个位置记录。

ID  Location
51699457    LOC000009723262
51699457    LOC000009844932
51699457    LOC000011336522
51699457    LOC000018023392
51699458    LOC000008089735
51699458    LOC000016725720
51699458    LOC000028801236
51699459    LOC000034119383
51699459    LOC000034901728
51699459    LOC000035640991
51699459    LOC000035658463

O / P

ID  Location
51699457    LOC000009723262
51699457    LOC000009844932
51699458    LOC000008089735
51699458    LOC000016725720
51699459    LOC000035640991
51699459    LOC000035658463

任何帮助都将不胜感激。

Answer 1

这很有用。

df.groupby('ID').head(2).reset_index(drop=True)