擅长:python、mysql、java
<p>您可以使用<code>numpy.where</code>执行此任务</p>
<pre><code>import numpy as np
a = np.array([[0, 1, 2, 3, 4, 5, 6, 7, 9, 8]])
b = np.where(a==np.max(a), a, 0)
print(b)
</code></pre>
<p>输出</p>
<pre><code>[[0 0 0 0 0 0 0 0 9 0]]
</code></pre>
<p>说明:这确实会创建与<code>a</code>形状相同的<code>numpy.array</code>,并从满足条件的<code>a</code>和所有其他位置的<code>0</code>创建相应的值</p>