У меня есть такой фрейм данных:
Name subname Feature1 Feature2 ...
AAA a 0.123 0.345 ...
AAA b 0.123 0.345 ...
BBB a 0.123 0.345 ...
BBB b 0.123 0.345 ...
Я хочу создать метки (добавив новый столбец):
Name subname Feature1 Feature2 ...Class
AAA a 0.123 0.345 ... 1
AAA b 0.123 0.345 ... 1
BBB a 0.123 0.345 ... 2
BBB b 0.123 0.345 ... 2
Чтобы я мог поместить данные в модель классификации, есть ли способ эффективно создавать эти метки? Я получил более 5000 строк, большое спасибо.
Name
? Если это такdf['Class'] = pd.factorize(df['Name'])[0] + 1
, если это факторизация2D
, вы можете использоватьnp.unique
сreturn_inverse
- person user3483203   schedule 22.07.2019