python分段函数 保留两位小数_python – Numpy Arrays上的分段函数

什么是在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优化的.