У меня есть три массива:
a = [0.4, 0.3, 0.3]
b = [0.5, 0.2, 0.3]
c = [0.2, 0.7, 0.1]
Теперь я вычисляю argmax:
np.argmax(a) = 0
np.argmax(b) = 0
np.argmax(c) = 1
Таким образом, у меня есть массив d = [1, 0, 0], потому что первый элементбудет иметь максимальное значение, все остальные элементы равны нулю.
Как это сделать?Спасибо