擅长:python、mysql、java
<h3><code>np.where</code></h3>
<pre><code>np.where(a > 0.5, 1, 0)
# array([0, 0, 0, 1, 1, 1])
</code></pre>
<hr/>
<h3>用<code>astype</code></h3>
<pre><code>(a > .5).astype(int)
# array([0, 0, 0, 1, 1, 1])
</code></pre>
<hr/>
<h3><code>np.select</code></h3>
<pre><code>np.select([a <= .5, a>.5], [np.zeros_like(a), np.ones_like(a)])
# array([ 0., 0., 0., 1., 1., 1.])
</code></pre>
<hr/>
<h3>特殊情况:<code>np.round</code></h3>
<p>如果数组值是介于0和1之间的浮动值,并且阈值为0.5,则这是最佳解决方案。</p>
<pre><code>a.round()
# array([0., 0., 0., 1., 1., 1.])
</code></pre>