非フラットインデックスを返すnumpy配列のArgmax

Question 1

Numpy配列の最大要素のインデックスを取得しようとしています。これは、を使用して実行できますnumpy.argmax。私の問題は、配列全体で最大の要素を見つけて、そのインデックスを取得したいということです。

numpy.argmax 私が望むものではない1つの軸に沿って適用することも、私が望むものの一種であるフラット化された配列に適用することもできます。

私の問題は、多次元インデックスが必要なときにnumpy.argmaxwithを使用するとaxis=Noneフラットインデックスが返されることです。

divmod平坦でないインデックスを取得するために使用できますが、これは醜い感じです。これを行うためのより良い方法はありますか？

Question 2

次numpy.unravel_index()の結果に使用できますnumpy.argmax()：

>>> a = numpy.random.random((10, 10))
>>> numpy.unravel_index(a.argmax(), a.shape)
(6, 7)
>>> a[6, 7] == a.max()
True

Question 3

np.where(a==a.max())

最大要素の座標を返しますが、配列を2回解析する必要があります。

>>> a = np.array(((3,4,5),(0,1,2)))
>>> np.where(a==a.max())
(array([0]), array([2]))

これは、と比較しargmaxて、最大値に等しいすべての要素の座標を返します。argmaxそれらの1つだけを返します（np.ones(5).argmax()returns 0）。

Question 4

最大値のすべての出現の非フラットインデックスを取得するには、次の代わりにを使用してeumiroの答えをわずかに変更できます。argwherewhere

np.argwhere(a==a.max())

>>> a = np.array([[1,2,4],[4,3,4]])
>>> np.argwhere(a==a.max())
array([[0, 2],
       [1, 0],
       [1, 2]])