В numpy.argmax функция разрыва между несколькими максимальными элементами такова, что возвращается первый элемент. Существует ли функциональность для рандомизации разрыва связи так, что все максимальные числа имеют равную вероятность выбора?
Ниже приведен пример непосредственно из документации numpy.argmax.
>>> b = np.arange(6)
>>> b[1] = 5
>>> b
array([0, 5, 2, 3, 4, 5])
>>> np.argmax(b) # Only the first occurrence is returned.
1
Я ищу способы, чтобы 1-й и 5-й элементы в списке возвращались с равной вероятностью.
Спасибо!