堆叠带偏移量的numpy数组

时间:2019-05-09 14:50:38

标签: python-3.x numpy

我正在尝试使用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]]

有什么主意吗?

0 个答案:

没有答案