擅长:python、mysql、java
<p>假设您使用numpy数组:</p>
<pre><code>import numpy as np
Scores = np.array([3,7,3,2,1,5,1])
animalIDs = np.array([1,2,3,4,5,6,7])
</code></pre>
<p>只需创建一个分数小于或等于2的布尔掩码</p>
<pre><code>animals2Delete = Scores <= 2
</code></pre>
<p>然后把这个面具贴在动物的身份证上:</p>
<pre><code>animalIDs[animals2Delete]
# returns array([4, 5, 7])
</code></pre>
<p>或者一步到位:</p>
<pre><code>animalIDs[Scores <= 2]
</code></pre>
<p>这不使用列表理解,而是numpys优化的迭代。结果,至少,应该是你想要的。你知道吗</p>