PD Próbka każdej klasy

df.groupby('classes').apply(lambda x: x.sample(sample_size))
Attractive Alpaca