zoukankan      html  css  js  c++  java
  • numpy.argmax()

    numpy.argmax(a, axis=None, out=None) 
    返回沿轴axis最大值的索引

    Parameters: 
    a : array_like                                  -数组
    axis : int, 可选                                -默认情况下,索引的是平铺的数组,否则沿指定的轴。 
    out : array, 可选                             -如果提供,结果以合适的形状和类型被插入到此数组中。 

    Returns: 
    index_array : ndarray of ints       -索引数组。它具有与a.shape相同的形状,其中axis被移除。 

    例子:

     1 >>> a = np.arange(6).reshape(2,3)
     2 >>> a
     3 array([[0, 1, 2],
     4        [3, 4, 5]])
     5 >>> np.argmax(a)
     6 5
     7 >>> np.argmax(a, axis=0)#0代表列
     8 array([1, 1, 1])
     9 >>> np.argmax(a, axis=1)#1代表行
    10 array([2, 2])
    11 >>>
    12 >>> b = np.arange(6)
    13 >>> b[1] = 5
    14 >>> b
    15 array([0, 5, 2, 3, 4, 5])
    16 >>> np.argmax(b) # 只返回第一次出现的最大值的索引
    17 1
  • 相关阅读:
    二月12日
    十日冲刺
    十日冲刺
    十日冲刺
    十日冲刺
    十日冲刺
    十日冲刺
    十日冲刺
    一周进度条博客
    十天冲刺
  • 原文地址:https://www.cnblogs.com/tectal/p/9461178.html
Copyright © 2011-2022 走看看