将函数应用于numpy中向量中的每个元素

时间:2014-12-12 20:29:16

标签: python numpy

我正在写一些性能关键代码..

def sig(x):
   return 1/(1+exp(x))

我有一个载体

x = [.... say n elements]

transform_x = [sig(ele) for ele in x]

transform_x = map(lambda ele: sig(ele), x) # bit faster than above loop

但有没有更好的方法来应用函数来实现上述效果,但可能采用矢量化方式(我认为应该更好)?

1 个答案:

答案 0 :(得分:4)

您可以将函数应用于整个数组,就像在MATLAB中一样:

transformed = sig(x)