我使用numpy网格网格获得的一组索引将两个n-d数组相乘。所以，我有两个名为current和previous的n-d数组，如果索引p和q，我有一个集合，如下所示：
time_lag = numpy.sum(current[p] * previous[q])
现在，我有另一个nd数组只填充0和1，我想要做的是确保这个乘法只发生在这个掩码设置为1的索引中。所以，目前我只是这样做： / p>
time_lag = numpy.sum(current[p] * previous[q] * mask[p])
这确保它将我不想要的区域归零，并且它们不会对总和做出贡献。我想知道是否有更好的方法只选择性地在那些掩模区域进行乘法运算？