擅长:python、mysql、java
<pre class="lang-py prettyprint-override"><code>>>> a = np.array([[0,1],[1,1],[2,1],[4,6],[10,3], [2,10]])
>>> a[:, 1] > 5 # Boolean mask
array([False, False, False, True, False, True])
>>> a[a[:, 1] > 5] # Filtered array
array([[ 4, 6],
[ 2, 10]])
>>> np.where(a[:, 1] > 5)[0] # Array of indices
array([3, 5])
</code></pre>