Maske ein 3D-Array mit einer 2d-Maske in numpy

Question

Ich habe ein 3-dimensionales Array, das ich mit einem 2-dimensionalen Array, das die gleichen Dimensionen wie die beiden am weitesten rechts von der 3-dimensionalen Array hat, maskieren möchte. Gibt es eine Möglichkeit, dies zu tun, ohne die folgende Schleife zu schreiben?

import numpy as np 

nx = 2 
nt = 4 

field3d = np.random.rand(nt, nx, nx) 
field2d = np.random.rand(nx, nx) 

field3d_mask = np.zeros(field3d.shape, dtype=bool) 

for t in range(nt): 
    field3d_mask[t,:,:] = field2d > 0.3 

field3d = np.ma.array(field3d, mask=field3d_mask) 

print field2d 
print field3d 

Answer 1

Ohne die Schleife könnte man es schreiben als:

field3d_mask[:,:,:] = field2d[np.newaxis,:,:] > 0.3 

Zum Beispiel:

field3d_mask_1 = np.zeros(field3d.shape, dtype=bool) 
field3d_mask_2 = np.zeros(field3d.shape, dtype=bool) 

for t in range(nt): 
    field3d_mask_1[t,:,:] = field2d > 0.3 

field3d_mask_2[:,:,:] = field2d[np.newaxis,:,:] > 0.3 

print((field3d_mask_1 == field3d_mask_2).all()) 

gibt:

Wahre

Answer 2

Es gibt numpy.broadcast_to (neu in Numpy 1.10.0):

field3d_mask = np.broadcast_to(field2d > 0.3, field3d.shape)