用于多维ndarray的argsort

用于多维ndarray的argsort,第1张

用于多维ndarray的argsort

上面的答案现在已经过时了,因为在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]])


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5644980.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存