Python NumPy：如何使用方程填充矩阵

Question

我想初始化一个矩阵A，对于某些f使用方程A_i,j = f(i,j)（这并不重要）。

如何简洁地避免出现两个for循环的情况？

Answer 1

numpy.fromfunction符合这里的法案。从文档

例子：

>>> import numpy as np 
>>> np.fromfunction(lambda i, j: i + j, (3, 3), dtype=int) 
array([[0, 1, 2], 
    [1, 2, 3], 
    [2, 3, 4]]) 

Answer 2

人们还可以让你的数组的索引与numpy.indices，然后在矢量时尚应用功能f，

import numpy as np 

shape = 1000, 1000 

Xi, Yj = np.indices(shape) 

A = (2*Xi + 3*Yj).astype(np.int) # or any other function f(Xi, Yj)