Y = np.array([[[11, 2], [3, 11], [-4, 100]], [[0, 11], [100, 3], [4, 11]],
    [[1, 4], [11, 100], [2, 3]], [[100, 3], [1, 1], [11, 1]]])
ind = X.argmax(axis=0)
a1, a2 = np.indices(ind.shape)
Найти максимум 3D np.array вдоль оси = 0