擅长:python、mysql、java
<p>如果您有最新的numpy(1.7),那么最好的方法是使用<a href="https://github.com/numpy/numpy/blob/6c729b4423857850e6553cf6c2d0fc8b026036dd/numpy/core/numeric.py#L2133" rel="nofollow">^{<cd1>}</a>,它将自动广播这些形状:</p>
<pre><code>import numpy as np
a = np.arange(10)
n = 5.000001
np.isclose(a, n).nonzero()
#(array([5]),)
</code></pre>
<p>或者,如果您只期望一个匹配:</p>
^{pr2}$
<p>(<code>np.nonzero</code>与<code>np.where</code>基本相同,只是它没有if条件then/else功能)</p>