Sample Pandas dataframe based on values in column

Question

I have a large dataframe that I want to sample based on values on the target column value, which is binary : 0/1

I want to extract equal number of rows that have 0's and 1's in the "target" column. I was thinking of using the pandas sampling function but not sure how to declare the equal number of samples I want from both classes for the dataframe based on the target column.

I was thinking of using something like this:

df.sample(n=10000, weights='target', random_state=1)

Not sure how to edit it to get 10k records with 5k 1's and 5k 0's in the target column. Any help is appreciated!

Answer 1

You can use DataFrameGroupBy.sample method as follwing:

sample_df = df.groupby("target").sample(n=5000, random_state=1)