/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

0 vote153 views1 answer
Convert matplotlib figure to cv2 image
Python
2 votes145 views1 answer
Python typing as int or float
Python
0 vote152 views1 answer
Dataframe show to semicolon python
Python
0 vote177 views1 answer
Return the number of elements in this RDD.
Python
1 vote146 views1 answer
Program to segregate positive and negative numbers in same list
Python
0 vote150 views1 answer
Python check if array is sorted
Python
1 vote142 views1 answer
Python detect if tkinter page closed
Python
1 vote144 views2 answers
Correlation between two columns pandas
Python
1 vote174 views3 answers
Recursive function
Python
14 votes157 views1 answer
Sort list of dictionaries by key python
Python