Mask 3D-array by 2D-array for slicing without for loop

Question

I have something like this

import numpy as np 

array_3D =  np.random.rand(3,3,3) 
array_2D = np.random.randint(0, 3 , (3,3)) 

for i in range(3):
    for j in range(3): 
        array_3D[:, i, j][:array_2D[i, j]]=np.nan

Is there a way to do this without the double for loop?

Divakar · Accepted Answer · 2020-09-24 14:06:32Z

1

Create the mask with outer ranged-comparison and then assign -

mask = np.less.outer(np.arange(len(array_3D)), array_2D)  
array_3D[mask] = np.nan

answered Sep 24, 2020 at 14:06

Divakar

222k19 gold badges273 silver badges374 bronze badges

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

Collectives™ on Stack Overflow

Mask 3D-array by 2D-array for slicing without for loop

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