numpy总结
这里写目录标题np.take_along_axis(a, ai, axis=1)
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_ax
原创
2020-12-22 10:05:59 ·
1907 阅读 ·
0 评论