np.take_along_axis(a, ai, axis=1)

>> np.sort(a, axis=1)
array([[10, 20, 30], [40, 50, 60]])
>>> ai = np.argsort(a, axis=1); ai
array([[0, 2, 1], [1, 2, 0]])
>>> np.take_along_axis(a, ai, axis=1)
array([[10, 20, 30], [40, 50, 60]])

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
 

文章来源: blog.csdn.net，作者：快了的程序猿小可哥，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/qq_35914625/article/details/111506838