python2.7中如何使用vectorize numpy函数？

Question

我在 ipython 中尝试了以下代码：我想使用矢量化并给出函数 prox(x,tau)。但是 lambda 中的第一个值总是出现两次。

In [32]: a = np.array([[ 1., 2.],[ 3., 4.]])
In [33]: def prox(x, tau):
    ...:     print x, tau
    ...:     if x >= tau:
    ...:         print "first"
    ...:         return x-tau
    ...:     if -tau <= x and x <= tau:
    ...:         print "second"
    ...:         return 0.0
    ...:     if x <= -tau:
    ...:         print "third"
    ...:         return x+tau

In [34]: b = np.vectorize(lambda x: prox(x, 2))(a[:,1:])
In [35]: b 
2.0 2
first
2.0 2
first
4.0 2
first

为什么在第 35 行两次打印相同的值？ 2.0 2

Answer 1

如果您不指定otypes，则vectorize 使用第一个值执行测试计算，并使用它来确定它返回的数组的dtype。因此对第一项进行双重评价。

通常，额外的计算并不重要。但小心点。如果初始计算返回一个整数（例如标量 0），则返回的数组也将是整数，在后续计算中会丢失任何浮点值。

有关更多详细信息，请查看vectorize 的文档。

vectorize is indeterminate - 意外的整数 otype 产生的错误。我在其他 SO 问题中看到过这个错误。