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

Question

我希望初始化一个矩阵A,使用A_i,j = f(i,j)某些方程f(这不重要).

我怎么能这样简洁地避免我有两个for循环的情况？

Answer 1

numpy.fromfunction适合这里的账单.

来自doc的示例:

>>> 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]])