在numpy中通过数组嵌套for循环更好吗？

Question

通常我需要遍历一个数组并对每个条目执行一些操作，其中操作可能取决于索引和条目的值。这里有一个简单的例子。在

import numpy as np

N=10
M = np.zeros((N,N))

for i in range(N):
    for j in range(N):
        M[i,j] = 1/((i+2*j+1)**2)

有没有一种更短、更干净或更像Python的方式来完成这些任务？在

Answer 1

我想说这是执行迭代的最直接和最普遍理解的方法。在

另一种方法是迭代这些值并为给定的（i，j）对调用函数

import itertools

N = 10
M = np.zeros((N,N))

def do_work(i, j):
       M[i,j] = 1/((i+2*j+1)**2)

[do_work(i, j) for (i, j) in itertools.product(xrange(N), xrange(N))]

我刚用过itertools.product要为可能的（i，j）值创建生成器，您也可以使用for循环。在

^{pr2}$