标签: python numpy
假设给出了一个不同且有序的p + 1实数的ndarray t = [t_0,t_1,...,t_p]。
我有另一个ndarray(更大)T = [T_0,T_1,...,T_N] N + 1个实数,都属于[t_0,t_p]。
我想要的是(唯一定义的)索引i_j的数组[i_0,i_1,...,i_N],使得T_j属于[t_i_j,t_i_ {j + 1} [。
如果有帮助的话,我可以做出T也会被排序的事情。我认为这样的基本操作可能是numpy库的一部分,但我似乎无法找到它。
答案 0 :(得分:0)
正如Paul Panzer在评论中说的那样,np.searchsorted(t, T)带有可选的side参数以照顾边界。
np.searchsorted(t, T)
side
文档为here。