我正在尝试编译一些函数,然后使用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中的索引从数组中删除一个元素?