Numpy.argmax()

numpy.argmax(a, axis=None, out=None)

返回沿軸axis最大值的索引。

Parameters:

  • a : array_like 數(shù)組
  • axis : int, 可選
    默認情況下，索引的是平鋪的數(shù)組，否則沿指定的軸。
  • out : array, 可選
    如果提供，結果以合適的形狀和類型被插入到此數(shù)組中。
  • Returns: index_array : ndarray of ints
    索引數(shù)組。它具有與a.shape相同的形狀，其中axis被移除。

示例:

>>> a
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argmax(a)
5
>>> np.argmax(a, axis=0)#0代表列
array([1, 1, 1])
>>> np.argmax(a, axis=1)#1代表行
array([2, 2])
>>>
>>> b = np.arange(6)
>>> b[1] = 5
>>> b
array([0, 5, 2, 3, 4, 5])
>>> np.argmax(b) # 只返回第一次出現(xiàn)的最大值的索引
1
最后編輯于 ：
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
平臺聲明：文章內容（如有圖片或視頻亦包括在內）由作者上傳并發(fā)布，文章內容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務。

推薦閱讀更多精彩內容

  • NumPy是Python中關于科學計算的一個類庫，在這里簡單介紹一下。 來源：https://docs.scipy...
    灰太狼_black閱讀 1,252評論 0贊 5
  • 基礎總結來源： https://www.toutiao.com/a6458488956890645...
    財務自由_lang閱讀 488評論 0贊 0
  • 先決條件 在閱讀這個教程之前，你多少需要知道點python。如果你想從新回憶下，請看看Python Tutoria...
    舒map閱讀 2,602評論 1贊 13
  • import numpy as np 創(chuàng)建ndarray data1 = [6,7.5, 8, 0, 1]arr1...
    陸文斌閱讀 726評論 0贊 1
  • 最近在寫個性化推薦的論文，經(jīng)常用到Python來處理數(shù)據(jù)，被pandas和numpy中的數(shù)據(jù)選取和索引問題繞的比較...
    shuhanrainbow閱讀 4,595評論 6贊 19