/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

2 votes326 views1 answer
Copy website python
Python
0 vote313 views1 answer
Numpy int64 to int
Python
45 votes332 views6 answers
Array python
Python
0 vote357 views1 answer
Python set current working directory debugging
Python
1 vote343 views1 answer
Spike python
Python
0 vote364 views1 answer
Ram clear in python
Python
0 vote335 views1 answer
Django.db.utils.ProgrammingError: can't adapt type 'dict_keys'
Python
18 votes341 views2 answers
Import local module python
Python
1 vote345 views6 answers
Code grepper
Python
82 votes314 views6 answers
Activate python venv in windows
Python