擅长:python、mysql、java
<p>作为另一种解决方案,<code>np.choose</code>对做出选择很有用。你知道吗</p>
<pre><code>>>> x = np.arange(16).reshape(4,4)
</code></pre>
<p>所以<code>x</code>看起来像这样:</p>
<pre><code>array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15]])
</code></pre>
<p>现在,在每列中的特定行<code>y</code>处选择值可以如下所示:</p>
<pre><code>>>> y = np.array([3, 0, 2, 1])
>>> np.choose(y, x)
array([12, 1, 10, 7])
</code></pre>