gpt4 book ai didi

python - 对二维数组使用 searchsorted

转载 作者:行者123 更新时间:2023-11-28 22:34:29 25 4
gpt4 key购买 nike

我想要像函数这样的索引:np.searchsorted([1,2,3,4,5], 3) 返回 2 但如果是二维维度:np.searchsorted([[1,2],[3,4],[5,6]], [5,6]) 应该返回 2。我该怎么做?

最佳答案

a = np.array([[1,2],[3,4],[5,6]])
b = np.array([5,6])
np.where(np.all(a==b,axis=1))

关于python - 对二维数组使用 searchsorted,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38948042/

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