擅长:python、mysql、java
<p>以下是您的数据:</p>
<pre><code>import numpy as np
arr = np.array([[ 0, 0, 0, 255, 0],
[ 0, 1, 0, 0, 255],
[ 0, 2, 0, 255, 0]])
a,b = 0,2 # [a,b] is what we are looking for, in the first two cols
</code></pre>
<p><strong>下面是获取包含[a,b]的行索引的解决方案:</strong></p>
^{pr2}$
<p><strong>输出:</strong></p>
<blockquote>
<p>2</p>
</blockquote>
<p><strong>说明:</strong></p>
<p>了解其工作原理的最佳方法是打印每个部分:</p>
<pre><code>print (arr[:,0]==[a])
</code></pre>
<p>输出:</p>
<blockquote>
<p>[ True True True]</p>
</blockquote>
<pre><code>print (arr[:,1]==[b])
</code></pre>
<p>输出:</p>
<blockquote>
<p>[False False True]</p>
</blockquote>
<pre><code>print (np.logical_and(arr[:,0]==[a],arr[:,1]==[b]))
# print (np.logical_and([ True True True], [False False True]))
</code></pre>
<p>输出:</p>
<p>[假-假-真]</p>