代码之家  ›  专栏  ›  技术社区  ›  dia

np.argsort文档中的示例

  •  0
  • dia  · 技术社区  · 8 年前

    因为某种原因,我不能解决这个问题。

    根据这里1维数组的例子, https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html

    x = np.array([3, 1, 2])
    np.argsort(x)
    

    数组([1,2,0])

    我自己也试过。但是默认情况下,返回结果应该是升序的..意思是

    x([result])
    

    退货

    数组([1,2,3])

    因此,结果不应是[2,0,1]

    我这里缺什么?

    1 回复  |  直到 8 年前
        1
  •  0
  •   ajsp    8 年前

    x = np.array([3, 1, 2])
    np.argsort(x)
    >>>array([1, 2, 0])
    

    array([1, 2, 0])

    argsort 值,然后用这些值索引初始数组:

    x   = np.array([3, 1, 2])
    srt_positions = np.argsort(x)
    
    for k in srt_positions:
        print x[k]
    
    >>> 1, 2, 3