从数组numba删除元素

时间:2019-03-11 13:53:46

标签: python arrays numba

我正在尝试编译一些函数,然后使用numba进行并行化。在我的功能之一中,我将基于索引删除一维数组的元素。数组可以是任意大小,但始终为一维。

@njit
def FormFullDesign2(Xcross,Xself,ct):
[Np,Mhc,Ncells,R] = Xcross.shape
cellC = np.arange(0,Ncells); cellC = np.delete(cellC,ct)

ct的范围是0到len(Ncells)。 恐怕创建布尔型掩码将不起作用,因为稍后我将使用cellC的内容来为数组建立索引:

    for cc in range(0,Ncells-1):
    Xfc = np.concatenate((np.squeeze(Xfc),np.squeeze(Xcross[:,:,cellC[cc],:])),axis=1)

那么有没有一种方法可以根据Numba中的索引从数组中删除一个元素?

0 个答案:

没有答案