Учитывая 2 2x2 массива Numpy, каждый элемент имеет значение от 0 до 1, я хотел бы найти один массив из 2, который имеет максимальное значение, и выполнить это сравнение для каждого элемента.Например, учитывая:
A = [[.6 .2] [.3 .4]] и B = [[.4 .5] [.7 .1]], я бы хотел что-то вроде: [[A B] [A B]] back.В идеале на выходе должно быть некоторое число, [[1 2] [1 2]], где 1 представляет A, а 2 представляет B. Таким образом, если я сравню, скажем, 10 массивов, на выходе будет целое число от 1 до 10 в качестве каждого элемента, чтоможет быть легко нанесен на график pcolor.
Если я просто объединю эти массивы в один 2x2x2 и выполню np.amax (комбинированный_кател, ось = 0), я получу максимальное значение, но не знаю какой массивэто происходит от.
Цель всего этого состоит в том, что каждый массив представляет категорию и содержит вероятности возникновения этой категории.Я хотел бы знать для каждой позиции элемента [0] [0], [0] [1], [1] [0] и [1] [1], какая категория наиболее вероятна в этой позиции.