擅长:python、mysql、java
<p>纯粹使用<code>numpy</code>:</p>
<pre><code>import numpy as np
daily_choices = np.array([['apple', 'orange'],['strawberry', 'orange'],['watermelon', 'apple']])
decision = np.array([0, 1, 0])
n_fruits = 2
fruit_range = np.reshape(np.arange(n_fruits), (-1, n_fruits))
indices = np.reshape(decision, (len(decision), 1)) == fruit_range
daily_choices[indices]
</code></pre>
<p>输出:</p>
<pre><code>array(['apple', 'orange', 'watermelon'], dtype='<U10')
</code></pre>