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])

浙公网安备 33010602011771号