上面的答案现在已经过时了,因为在numpy
1.15中添加了新功能以使其更简单。take_along_axis(https://docs.scipy.org/doc/numpy-1.15.1/reference/generation/numpy.take_along_axis.html)允许您执行以下 *** 作:
>>> a = np.array([[3,1,2],[8,9,2]])>>> np.take_along_axis(a, a.argsort(axis=-1), axis=-1)array([[1 2 3] [2 8 9]])
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)