1 |
test_df |
номер | город | колич. признак | |
---|---|---|---|
0 | 2 | москва | 1 |
1 | 3 | питер | 2 |
2 | 4 | саратов | 3 |
3 | 5 | москва | 4 |
4 | 6 | питер | 3 |
5 | 7 | москва | 4 |
6 | 8 | сыктывкар | 5 |
7 | 9 | саратов | 6 |
1 2 |
df_nonbinary = pd.get_dummies(test_df['город'], ) df_nonbinary |
москва | питер | саратов | сыктывкар | |
---|---|---|---|---|
0 | 1.0 | 0.0 | 0.0 | 0.0 |
1 | 0.0 | 1.0 | 0.0 | 0.0 |
2 | 0.0 | 0.0 | 1.0 | 0.0 |
3 | 1.0 | 0.0 | 0.0 | 0.0 |
4 | 0.0 | 1.0 | 0.0 | 0.0 |
5 | 1.0 | 0.0 | 0.0 | 0.0 |
6 | 0.0 | 0.0 | 0.0 | 1.0 |
7 | 0.0 | 0.0 | 1.0 | 0.0 |
1 2 |
new_df = pd.concat([test_df, df_nonbinary], axis=1 ) new_df |
номер | город | колич. признак | москва | питер | саратов | сыктывкар | |
---|---|---|---|---|---|---|---|
0 | 2 | москва | 1 | 1.0 | 0.0 | 0.0 | 0.0 |
1 | 3 | питер | 2 | 0.0 | 1.0 | 0.0 | 0.0 |
2 | 4 | саратов | 3 | 0.0 | 0.0 | 1.0 | 0.0 |
3 | 5 | москва | 4 | 1.0 | 0.0 | 0.0 | 0.0 |
4 | 6 | питер | 3 | 0.0 | 1.0 | 0.0 | 0.0 |
5 | 7 | москва | 4 | 1.0 | 0.0 | 0.0 | 0.0 |
6 | 8 | сыктывкар | 5 | 0.0 | 0.0 | 0.0 | 1.0 |
7 | 9 | саратов | 6 | 0.0 | 0.0 | 1.0 | 0.0 |