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)