Numpy Arrays上的分段函数

Question

什么是在Numpy数组上应用分段函数的有效(速度)方法？

比方说,例如,分段函数就像

For (1) :  x<=2 f(x) = 2*x + x^2
    (2) :  x>2  f(x) = -(x^2 + 2)

这就是我做的.

data = np.random.random_integers(5, size=(5,6))
print data
np.piecewise(data, [data <= 2, data > 2],
             [lambda x: 2*x + pow(2, x),
              lambda x: -(pow(x, 2) + 2)])

data = 
[[4 2 1 1 5 3]
 [4 3 3 5 4 5]
 [3 2 4 2 5 3]
 [2 5 4 3 1 4]
 [5 3 3 5 5 5]]
output = 
array([[-18,   8,   4,   4, -27, -11],
       [-18, -11, -11, -27, -18, -27],
       [-11,   8, -18,   8, -27, -11],
       [  8, -27, -18, -11,   4, -18],
       [-27, -11, -11, -27, -27, -27]])

对于较小的数组,大型数组,许多函数等,是否有一种有效的方法？我担心的是使用lambda函数.不确定这些是否是Numpy优化的.

Answer 1

在这种情况下，您不应该担心 lambda：Numpy 优化是通过让函数批量同时计算多个值来减少调用开销。在每次调用 时np.piecewise，每个函数funclist， （函数部分）因此，这些 lambda 是以 numpy 优化的方式调用的。

类似的是np.select（并且np.where恰好有两个部分）。调用开销与以相同方式矢量化相同，但它将评估所有数据点的所有函数。因此，它会比 慢np.piecewise，特别是当函数很昂贵时。在某些情况下，它更方便（没有 lambda），并且可以更轻松地将这一概念扩展到许多变量。