如何仅使用argmin（）对数组进行排序

Question

如何仅使用argmin()？

按升序对数组进行排序

例如，如果我定义sort_function(x)仅使用x返回argmin()的升序，并执行：

print(sort_function(np.array([1,6,3,7]))

结果应为[1 3 6 7]。

Answer 1

这里是快速的O(N ** 2) -

1. 在列表中调用np.argmin
2. 在该索引处弹出元素
3. 重复直到列表为空

def sort_function(x):
    y = list(x)
    while y:
        yield y.pop(np.argmin(y))

list(sort_function(np.array([1, 6, 3, 7])))
[1, 3, 6, 7]