擅长:python、mysql、java
<p>掩码数组的实现正是为了您的目的。在</p>
<p>表现与斯文的回答相同:</p>
<pre><code>height_r_t = np.ma.masked_where(repelling_force_prefactor == 0, height_r_t)
repelling_forces = np.ma.exp(-(height_r_t/potential_steepness))
</code></pre>
<p>遮罩数组的优点是不必对数组进行切片和扩展,其大小始终相同,但numpy自动知道在遮罩数组的情况下不计算exp。在</p>
<p>此外,您可以使用不同的遮罩求和阵列,结果阵列有遮罩的交集。在</p>