Enmascara una matriz 3d con una máscara 2d en números

Tengo una matriz tridimensional que quiero enmascarar usando una matriz bidimensional que tiene las mismas dimensiones que las dos más a la derecha de la matriz tridimensional. ¿Hay una manera de hacer esto sin escribir el siguiente bucle?

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 

Sin el bucle podrías escribirlo como:

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

Por ejemplo:

 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()) 

da:

Cierto

Hay numpy.broadcast_to (nuevo en Numpy 1.10.0):

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