我有一个 3 维数组,我想使用一个 2 维数组来屏蔽它,该数组的维度与 3 维数组最右边的两个数组的维度相同。有没有办法在不编写以下循环的情况下执行此操作?
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
There's numpy.broadcast_to http://docs.scipy.org/doc/numpy-1.10.0/reference/generated/numpy.broadcast_to.html(Numpy 1.10.0 中的新功能):
field3d_mask = np.broadcast_to(field2d > 0.3, field3d.shape)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)