У меня есть матрица shape=(3,3), а строки представляют распределения вероятностей. Вот пример:
distribution_probablities=np.array([[1/3,2/3,0],
[0.5,0.2,0.3],
[1,0,0]])
Я хочу получить вектор длиной 3, и каждое значение представляет собой индекс выбранной вероятности. Возможный результирующий вектор таков: result=[1,0,0]
В первой строке у второго элемента больше шансов быть выбранным, во второй - то же самое для второго элемента и так далее. Другой возможный результат: result=[0,2,0]
. Можно ли сделать это без цикла for по строкам?