Выбор первых нескольких строк на основе постоянных значений столбцов

Question

Я хочу получить записи местоположения (первые 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)