如何仅使用argmin()对数组排序

2024-04-16 17:42:07 发布

您现在位置:Python中文网/ 问答频道 /正文

如何仅使用argmin()按升序对数组排序?你知道吗

例如,如果我定义了sort_function(x),它仅通过使用argmin()返回x的升序并执行以下操作:

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

结果应该是[1 3 6 7]。你知道吗


Tags: 定义排序npfunction数组sortarrayprint
1条回答
网友
1楼 · 发布于 2024-04-16 17:42:07

就在这里,在炽热的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]

相关问题 更多 >