Python random sample from dataframe with given characteristics

Question

I have a dataframe df with patients subject_id, including their gender and their age.

I would like to draw a random sample of size n from this dataframe, with the following characteristics:

50% male, 50% female
Median age of 40 years

Any idea how I could accomplish that using python? Thank you!

Does this answer your question? Random Sample of a subset of a dataframe in Pandas — Yoshikage Kira
– Yoshikage Kira, Commented Dec 1, 2021 at 8:54

Josh Friedlander · Accepted Answer · 2021-12-01 09:04:51Z

1

I think what you want is a little bit more complex than what DataFrame.sample provides out of the box. A random sample satisfying each of your conditions could be generated (respectively) like this:

Filter for women only, and randomly sample n/2, then do the same for men, and then pool them
Filter for under 40s, randomly sample n/2, then do the same for over-40s and then combine them. (Though note that this does not guarantee a median of exactly 40.)

If you want to combine the two constraints, you might need to sample 4 times - women under 40, men under 40, etc. But this is the general idea.

Code for sampling would look like:

df.loc[df.age < 40, 'subject_id'].sample(n/2)
df.loc[df.gender == 'F', 'subject_id'].sample(n/2)

answered Dec 1, 2021 at 9:04

Josh Friedlander

11.8k7 gold badges42 silver badges89 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Python random sample from dataframe with given characteristics

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related