/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 votes199 views1 answer
Reverse order np array
Python
0 vote183 views1 answer
Alpaca examples
Python
0 vote238 views1 answer
Add to middle of list python
Python
0 vote185 views1 answer
Difference between calling a function and referencing a function python
Python
56 votes184 views6 answers
How to open a file with python
Python
2 votes212 views2 answers
Django template array index
Python
0 vote189 views1 answer
Pandas create column if equals
Python
0 vote232 views1 answer
Size of variable
Python
2 votes186 views2 answers
Install anaconda python 2.7 and 3.6
Python
0 vote184 views1 answer
Python kill process by name
Python