numpy_03 排序

import numpy as np

if __name__ == '__main__':
    #排序
    array=np.array([[4.2,6.5,0.4],[2.,9.5,8.3]])
    # print(np.sort(array))#默认用axis=-1 的轴!
    # print(np.sort(array,axis=0))

    #获得排序的位置,结果是当前位置中放原来的哪个位置的数值!
    # print(np.argsort(array))#默认按axis=-1

    #复合排序
    array=np.array([[1,0,6],
                    [1,7,0],
                    [2,3,1],
                    [2,4,0]])

    # np.lexsort(a,b), sort by b,then by a
    index=np.lexsort([-1*array[:,0],array[:,2]]) #先按2列升序,再按0列降序
    print(index)
    print(array[index])

  

posted @ 2019-07-17 23:07  刘天睿  阅读(63)  评论(0)    收藏  举报