/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

3 votes179 views1 answer
How to get number of cores in python
Python
0 vote124 views1 answer
Fibonacci sequence in python
Python
0 vote172 views1 answer
Python map function using lambda function as one of the parameters
Python
0 vote163 views1 answer
Roll a dice
Python
0 vote166 views1 answer
Python slice a list a specific amount of times
Python
2 votes175 views1 answer
Remove comma from string python column
Python
0 vote171 views1 answer
Python capitilize
Python
0 vote162 views1 answer
No module named 'selenium.webdriver.common.action_chain'
Python
0 vote174 views1 answer
Selenium if statement python
Python
0 vote243 views1 answer
How to get table schema sql pyodbc
Python