/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

4 votes160 views3 answers
Print variable string python
Python
0 vote204 views1 answer
Condtion for equal time in selenium python
Python
0 vote181 views1 answer
How to make python do something every x seconds
Python
1 vote156 views2 answers
Where
Python
0 vote152 views1 answer
Python get desktop environment
Python
13 votes177 views2 answers
Pip version command
Python
2 votes154 views1 answer
Sqlalchemy query bilter by current month
Python
5 votes157 views2 answers
How to make a calcukatir un python
Python
0 vote140 views1 answer
'numpy.ndarray' object has no attribute 'count'
Python
1 vote145 views2 answers
How to add watermark in mp4 video using python
Python