/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

5 votes143 views3 answers
Debugging python
Python
2 votes147 views2 answers
Random choice dictionary python
Python
5 votes198 views4 answers
Dropna for specific column pandas
Python
-1 vote140 views2 answers
How to set pandas dataframe as global
Python
0 vote154 views1 answer
Np.linalg.eigvals positive check python
Python
1 vote149 views1 answer
How to count the occurrence of a word in string python
Python
0 vote175 views1 answer
Somma array python
Python
3 votes144 views1 answer
How to close the window in pygame
Python
2 votes164 views2 answers
How to read then overwrite a file with python
Python
0 vote157 views1 answer
Sort first element reverse sort second python
Python