Argmax ของอาร์เรย์จำนวนนับที่ส่งคืนดัชนีที่ไม่ใช่แบบแบน

ฉันพยายามหาดัชนีขององค์ประกอบสูงสุดในอาร์เรย์ Numpy ซึ่งสามารถทำได้โดยใช้numpy.argmax. ปัญหาของฉันคือฉันต้องการหาองค์ประกอบที่ใหญ่ที่สุดในอาร์เรย์ทั้งหมดและรับดัชนีของสิ่งนั้น

numpy.argmax สามารถใช้กับแกนเดียวซึ่งไม่ใช่สิ่งที่ฉันต้องการหรือบนอาร์เรย์แบบแบนซึ่งเป็นสิ่งที่ฉันต้องการ

ปัญหาของฉันคือการใช้numpy.argmaxกับaxis=Noneส่งคืนดัชนีแบนเมื่อฉันต้องการดัชนีหลายมิติ

ฉันสามารถใช้divmodเพื่อรับดัชนีที่ไม่แบน แต่รู้สึกน่าเกลียด มีวิธีที่ดีกว่านี้หรือไม่?

คุณสามารถใช้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

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

ส่งคืนพิกัดขององค์ประกอบสูงสุด แต่ต้องแยกวิเคราะห์อาร์เรย์สองครั้ง

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

สิ่งนี้เมื่อเปรียบเทียบกับจะargmaxส่งกลับค่าพิกัดขององค์ประกอบทั้งหมดที่เท่ากับค่าสูงสุด argmaxส่งคืนเพียงหนึ่งในนั้น ( np.ones(5).argmax()ผลตอบแทน0)

เพื่อให้ได้ดัชนีที่ไม่แบนของค่าสูงสุดที่เกิดขึ้นทั้งหมดคุณสามารถแก้ไขคำตอบของ eumiro เล็กน้อยโดยใช้argwhereแทนwhere:

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]])