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

Question

我希望使用等式A对某些A_i,j = f(i,j)初始化矩阵f（这并不重要）。

我怎样才能简明扼要地避免我有两个for循环的情况？

Answer 1

numpy.fromfunction符合此处的法案。

doc：

中的示例

XYZ

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)