gpt4 book ai didi

python - 沿 numpy 数组轴的 k-最小值的索引

转载 作者:太空狗 更新时间:2023-10-30 02:22:46 27 4
gpt4 key购买 nike

有没有办法在不使用循环的情况下沿 numpy 数组的轴返回 k-最小值的索引?

最佳答案

import numpy as np
x = np.array([[5, 2, 3],[1, 9, 2]]) # example data
k = 2 # return the indices of the 2 smallest values
np.argsort(x, axis=1)[:,0:k] # by row

array([[1, 2],
[0, 2]])

关于python - 沿 numpy 数组轴的 k-最小值的索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/8669261/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com