如何根據爲每個元素調用的函數製作numpy數組？

Question

的numpy.fromfunction使得與參數作爲陣列的功能的單個調用，如描述在例如this answer。所以要使用numpy.from函數，需要該函數接受數組參數並返回一個數組。

如何使numpy的陣列基於稱爲對於每個元件，其中函數返回標爲元件的功能，如例如使用max函數像下面？

z = numpy.fromfunction(max, (2, 2), dtype=int) 

這將然後返回：

[[0 1] 
[1 1]] 

Answer 1

您可以使用numpy.vectorize爲

vmax = numpy.vectorize(max) 
z = numpy.fromfunction(vmax, (2, 2), dtype=int) 

請雖然注意從文檔這樣一句話：

矢量化功能主要由fo提供方便，而不是表現。實現本質上是一個for循環。

就其性質而言numpy.vectorize會很慢。在這個例子中numpy.maximum將快得多：

z = numpy.fromfunction(numpy.maximum, (2, 2), dtype=int)