/Python/Pandas resamples stratified by columns values

Pandas resamples stratified by columns values

# stratified split dataset by column
from sklearn.model_selection import train_test_split
df_sample, df_drop_it = train_test_split(df, train_size =0.2, stratify=df['country'])

Source: datascience.stackexchange.com

Similar codes

47 votes219 views6 answers
Pandas read csv
Python
4 votes155 views3 answers
Pandas remove time from datetime
Python
0 vote201 views1 answer
How to find most repeated word in a string in python
Python
3 votes193 views2 answers
How to change role permissions in discord.py
Python
0 vote200 views1 answer
Date component
Python
0 vote194 views1 answer
While importing we detected an older version of numpy in
Python
0 vote162 views1 answer
Pyqt5 book
Python
1 vote206 views1 answer
Django print settings
Python
0 vote205 views1 answer
Concatenate strings of numpy array python
Python
0 vote202 views1 answer
Numpy randomly swap lines
Python