如何根据为每个元素调用的函数制作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)