gpt4 book ai didi

python - 切片上的 numpy searchsorted 词典

转载 作者:太空宇宙 更新时间:2023-11-04 03:47:45 35 4
gpt4 key购买 nike

假设我有一个已经排序的 numpy 数组。

>>> x
array([[1, 2, 3, 1],
[2, 2, 3, 2],
[2, 3, 3, 4],
[3, 4, 4, 4],
[4, 5, 5, 5]])

我想根据前列进行右手二分。

类似于 (quasicode) numpy.searchsorted(x, [2, 2], side='right') 应该返回 2,[2, 2, 3 之间的索引, 2][2, 3, 3, 4]

最佳答案

这个怎么样:

>>> i1 = np.searchsorted(x[:, 0], 2, side='l')
>>> i2 = np.searchsorted(x[i1:, 0], 2, side='r')
>>> i3 = np.searchsorted(x[i1:i1+i2, 1], 2, side='r')
>>> i1 + i3
2

关于python - 切片上的 numpy searchsorted 词典,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/22848290/

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