/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 votes259 views1 answer
How to change background color in python turtle
Python
8 votes247 views2 answers
How to create a random number between 1 and 10 in python
Python
0 vote239 views1 answer
How to access range of tuples in python
Python
2 votes281 views4 answers
Decision tree classifier sklearn
Python
0 vote269 views1 answer
Check django channels with redis setup
Python Django
0 vote272 views1 answer
Free download django app for windows 10
Python Django
0 vote245 views1 answer
Python select from list by boolean list
Python
1 vote247 views1 answer
Python discord discord.py disable remove help command
Python
1 vote211 views1 answer
Python tkinter change color of main window
Python
0 vote287 views1 answer
Python flask models user
Python Flask