擅长:python、mysql、java
<p>假设您希望,对于每一行,类0的概率最高的数组的索引:</p>
<pre><code>which = 0
np.stack([predict_prob1, predict_prob2, predict_prob3], axis=2)[:, which, :].argmax(axis=1)
</code></pre>
<p>输出:</p>
^{pr2}$
<p>对于1级:</p>
<pre><code>array([2, 1, 1, 1])
</code></pre>