擅长:python、mysql、java
<p>按照这些思路,应该可以有效地实现这一点,尽管目前还无法进行测试:</p>
<pre><code>rows = len(prediction)
p = prediction[np.arange(rows), catIndex.flatten()]
catResult = np.empty(rows, 'S1').fill('n')
catResult[p] = categories[catIndex.flatten()][p]
</code></pre>