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

Question

我希望初始化矩陣A ，對於某些f使用等式A_i,j = f(i,j) （這不重要）。

我怎么能這樣簡潔地避免我有兩個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]])

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)