Python NumPy : How to fill a matrix using an equation

Question

我想初始化一个矩阵 A，使用等式 A_i,j = f(i,j) 得到一些 f(它是这是什么并不重要)。

我怎样才能简洁地避免出现两个 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]])