apply a function on multiple dimension in numpy array

Question

I have a numpy array od dimension (343, 343, 250). It's an image time serie, with the first dim being x coordinate, 2nd dim y coordinate, and 3rd dim the date.

I try o apply a function on every pixel time serie as such:

def f(d):

   #do stuff with d
   return stuff

result = np.zeros((data.shape[0], data.shape[1]), dtype=cp.float32)
for i in range(data.shape[0]):
    for j in range(data.shape[1]:

        d = data[i,j,:]
        result[i,j] = f(d)

How could I do the same thing in a more numpy way (without the for loop) ?

Elliot Su · Accepted Answer · 2021-07-01 08:10:14Z

4

Using numpy.apply_along_axis:

import numpy as np
result = np.apply_along_axis(f, axis=2, arr=data)

answered Jul 1, 2021 at 8:10

Elliot Su

1761 silver badge6 bronze badges

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

Collectives™ on Stack Overflow

apply a function on multiple dimension in numpy array

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