我正在尝试使用np.array
将一些值定位到searchsorted
中,并且我希望得到的结果数组以自身为模的偏移量堆叠起来。
我可以这样:
import numpy as np
a = np.array([(1, 3.5), (1, 2.1), (1, 5.8), (1, 0.)])
b = np.arange(0.5, 5.5, 1.)
c = np.searchsorted(b, a[:, 1])
d = np.column_stack((c, c + 1))
但我想更直接地做到这一点,类似于:
c = np.column_stack((np.searchsorted(b, a[:, 1]), np.searchsorted(b, a[:, 1]) + 1))
无需重复调用np.searchsorted
。
对于上面的示例,结果应为:
[[3 4]
[2 3]
[5 6]
[0 1]]
有什么主意吗?