使用numpy的fromfunction的

Question

即時試圖使用fromfunction創建具有畝= 3和sig的高斯值的5x5矩陣= 2，這是我的嘗試：

from random import gauss 
import numpy as np 
np.fromfunction(lambda i,j: gauss(3,2), (5, 5)) 

這是結果：5.365244570434782

據我所知，從docs這應該工作，但我得到一個標量而不是5x5矩陣...爲什麼？以及如何解決這個問題？

Answer 1

numpy.fromfunction文檔極具誤導性。而不是重複調用你的函數，並從結果中建立一個數組，fromfunction實際上只會對你傳遞的函數進行一次調用。在這一次調用中，它將一些索引數組傳遞給你的函數，而不是單獨的索引。

剔除文檔字符串，執行如下：

def fromfunction(function, shape, **kwargs): 
    dtype = kwargs.pop('dtype', float) 
    args = indices(shape, dtype=dtype) 
    return function(*args,**kwargs) 

這意味着，除非你的函數廣播，numpy.fromfunction不會做像什麼文件說它做任何事情。