Я пытаюсь преобразовать массив вероятностей NumPy с суммой до 1,0 в двоичный массив.По сути, я хочу превратить максимальную вероятность в массиве в 1, а остальные в 0.Конечно, я знаю, как это сделать с помощью Python, но я надеялся, что NumPy или SciPy уже придумали встроенную функцию для повышения производительности операции (мы говорим о матрице, содержащей миллионы таких массивов).
Сравните предметы с максимальным значением и приведите к int
>>> a=np.array([0.435,0.24,0.241,0.13,0.56]) >>> np.int8( a == a.max() ) array([0, 0, 0, 0, 1], dtype=int8)