我们可以利用NumPy广播来实现向量化解决方案,只需将起始和结束索引与覆盖列长度的范围数组进行比较,就可以得到一个掩码,表示需要分配为1s的输出数组中的所有位置。 因此,解决方案应该是这样的 - ncols = z.shape[1] r = np.arange(z.shape[1]) mask = (index[:,0,None] <= r) & (index[:,1,None]...