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

Question

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

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

但是有没有更好的方法来应用该功能来实现上述效果，但可能是矢量化的（我认为应该更好）？

Answer 1

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

transformed = sig(x)