我正在使用 numpy 數(shù)組我遇到了障礙，也許你可以幫助我。所以我有一個二維 numpy 數(shù)組，這個二維數(shù)組中的每個數(shù)組都有一個最大值，我需要的是獲取最大值的索引而不是二維數(shù)組的索引，而是像使用一維數(shù)組一樣使用索引，我知道我可以用它numpy.argmax來查找索引，但索引是根據(jù)二維數(shù)組而不是一維數(shù)組，也許一個例子會更好：import numpy as nparr =  np.array([[512, 523, 491],                  [512, 531, 495]])index = np.argmax(arr, axis = 1)index2 = np.argmax(arr)print(index)print(index2) index = [1, 1] index2 = 4我的意思是它做了它應該做的事情，但是如果我需要像一維數(shù)組一樣的索引怎么辦？這是我想要的輸出：index = [1, 4] 所以第一個最大值是 523，所以第一個索引是 1，第二個最大值是 531，所以第二個索引是 4，就像它是一維數(shù)組一樣。也許這是一個新手問題，但我不確定該怎么做，這只是一個例子，數(shù)組arr可以更大，所以，任何幫助將不勝感激，謝謝！