擅长:python、mysql、java
<p>只需使用<code>np.argwhere</code></p>
<p>例如:</p>
<pre><code>>>> import numpy as np
>>> a = np.zeros((50, 50))
>>> a[3, 4] = 1
>>> a[4, 5] = 1
>>> for i, j in np.argwhere(a == 1): print(i, j)
...
(3, 4)
(4, 5)
</code></pre>
<p>(上面的示例是使用python 2运行的,因此显示了输出中的元组,但在其他方面与python 3中完全相同。)</p>