Applying a function element-wise to multiple numpy arrays

Question

Say I have two numpy arrays of the same dimensions, e.g.:

a = np.ones((4,))
b = np.linspace(0,4,4)

and a function that is supposed to operate on elements of those arrays:

def my_func (x,y):
   # do something, e.g.
   z = x+y
   return z

How can I apply this function to the elements of a and b in an element-wise fashion and get the result back?

You need to provide more data about your function. As is the answer is trivially a+b. — Daniel
– Daniel, Commented Apr 3, 2014 at 19:35

Eelco Hoogendoorn · Accepted Answer · 2014-04-03 19:36:00Z

1

It depends, really. For the given function; how about 'a+b', for instance? Presumably you have something more complex in mind though.

The most general solution is np.vectorize; but its also the slowest. Depending on what you want to do, more clever solutions may exist though. Take a look at numexp for example.

answered Apr 3, 2014 at 19:36

Eelco Hoogendoorn

10.8k1 gold badge46 silver badges43 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Applying a function element-wise to multiple numpy arrays

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related