您的位置：首页 > 其它

Tricks（二十七）—— 如何在不使用 np.argsort 的情况下获得排序后元素的下标

2016-04-09 23:30 344 查看
首先来看使用 np.argsort 得到下标：

>>> x = [1, 3, 2]
>>> np.argsort(x)
array([0, 2, 1], dtype=int32)


如何不使用 np.argsort 的情况依然获得排序后元素所在的下标呢？

通过 enumerate，为序列中的每一个元素增加一个维度，也即绑定下标：

>>> [x[0] for x in sorted(enumerate(x), key=lambda x: x[1])]
[0, 2, 1]
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 点击这里给我发消息
标签： 
相关文章推荐
新的分享
章节导航