python:numpy数组的条件加运算符

我想在两个 numpy 数组上应用加号运算符。但是，有一个限制。 我只想对数组一中的元素不为零的元素求和。我可以循环执行此操作，但这非常慢。有没有典型的 numpy 方法？

if a1[xyz] != 0: 
    r[xyz] = a1[xyz] + a2[xyz]  

请您参考如下方法:

使用面具:

mask = a1 != 0 
r[mask] = a1[mask] + a2[mask] 

这假设 r, a1, a2 具有相同的形状。