擅长:python、mysql、java
<p>您可能想要使用的是<a href="http://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html" rel="nofollow">^{<cd1>}</a>函数。
这样使用:</p>
<pre><code> >>>unicorns=np.array([[1, "black", 0.0, 'Pinky', 1] ,
[2, "black", 0.0, 'Winky', 1],
[3, "white", 0.0, 'Lala', 1],
[4, "white", 0.0, 'Merlin', 1],
[5, "black", 0.0, 'Meriva', 1],
[6, "white", 0.0, 'Panda', 1]])
>>> np.where(unicorns[:,1] == "black")
(array([0, 1, 4]),)
>>> unicorns[np.where(unicorns[:,1] == "black")]
array([['1', 'black', '0.0', 'Pinky', '1'],
['2', 'black', '0.0', 'Winky', '1'],
['5', 'black', '0.0', 'Meriva', '1']],
dtype='|S8')
</code></pre>