Карта Numpy -категориальные данные на вектор Numpy ⇐ Python

[Anonymous]

У меня есть массив Numpy, который выглядит как: < /p>

my_arr = array([[0., 0., 0., 0., 1., 0.],
[0., 1., 0., 0., 0., 0.],
[0., 0., 0., 1., 0., 0.],
[0., 0., 0., 0., 1., 0.],
[1., 0., 0., 0., 0., 0.],
[0., 0., 0., 0., 1., 0.],
[0., 1., 0., 0., 0., 0.],
[0., 0., 0., 0., 1., 0.],
...
...]
< /code>

Я хочу вернуть вектор, который будет содержать для каждого вектора my_arr Индекс входа со значением One. Как я могу это сделать?

Подробнее здесь: https://stackoverflow.com/questions/518 ... mpy-vector