pandas get first n rows in group code example

Example: select first row of every group pandas

df.drop_duplicates(subset='A')