对元素排序并取前n小值的index

ipdb> m                                                                                               
array([[ 3,  3,  3, 15],
       [ 3, 18, 17, 16],
       [ 5, 12, 17, 10]])
ipdb> np.argsort(m, axis = 0)                                                                         
array([[0, 0, 0, 2],
       [1, 2, 1, 0],
       [2, 1, 2, 1]])
ipdb> np.argsort(m, axis = 0)[0 : 1]                                                                  
array([[0, 0, 0, 2]])
ipdb> np.argsort(m, axis = 0)[0 : 1, :]                                                               
array([[0, 0, 0, 2]])
ipdb> np.argsort(m, axis = 0)[0:2, :]                                                                 
array([[0, 0, 0, 2],
       [1, 2, 1, 0]])
ipdb> np.argsort(m, axis = 0)[:,0:2]                                                                  
array([[0, 0],
       [1, 2],
       [2, 1]])