Np argmax多個最大值

np.argmax() 函式是 Python 的 NumPy 庫中的一個函式,用於返回數組中最大值的索引。但是,如果你想要找到多個最大值的索引,你可以使用 np.argpartition() 函式。

以下是一個簡單的例子,展示了如何找到數組中前三個最大值的索引:

import numpy as np

# 創建一個數組
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

# 找到前三個最大值的索引
inds = np.argpartition(arr, -3)[-3:]

# 列印結果
print(inds)

在這個例子中,np.argpartition() 函式的第一個參數是數組 arr,第二個參數 -3 表示我們要找到前三個最大值的索引。然後我們使用 [-3:] 切片來獲取索引的最後一個部分,即前三個最大值的索引。

請注意,np.argpartition() 函式不會對數組進行排序,它只會返回一個分區後的數組,其中包含我們指定的索引。如果你想要對數組進行排序,你可以使用 np.argsort() 函式,但是這個函式會返回完全排序的數組,而不是只包含最大值的索引。