擅长:python、mysql、java
<p>那<code>np.ndindex</code>怎么样?在</p>
<pre><code>np.ndindex(1000,1000)
</code></pre>
<p>这将返回一个iterable对象:</p>
^{pr2}$
<p>通常,如果您有一个数组,您可以通过以下方式构建索引iterable:</p>
<pre><code>index_iterable = np.ndindex(*arr.shape)
</code></pre>
<p>当然,总有<code>np.ndenumerate</code>可以这样实现:</p>
<pre><code>def ndenumerate(arr):
for ix in np.ndindex(*arr.shape):
yield ix,arr[ix]
</code></pre>