CODEWITHSUNDEEP
pythonpandas
Manny
Manny

Reputation: 87

list first N rows from Pandas df using groupby

I am using the following code to group a pandas df, and pass the values grouped to a list

  df.groupby(['Column1','Column2'])['Column3'].apply(list)

This returns all values from column3 in the list.
How can i get the list to include only first N examples only from column3 ?

Upvotes: 0

Views: 184

Answers (1)

jezrael
jezrael

Reputation: 862571

Use indexing with lambda function:

df.groupby(['Column1','Column2'])['Column3'].apply(lambda x: list(x)[:N])

Or:

df.groupby(['Column1','Column2'])['Column3'].apply(lambda x: list(x.iloc[:N]))

EDIT:

df.groupby(['Column1','Column2'])['Column3'].apply(lambda x: list(x.unique()[:N]))

Upvotes: 1

Related Questions