web-dev-qa-db-ja.com

numpyの2Dマスクで3D配列をマスクする

3次元配列の右端の2つと同じ次元の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
pythonarraysnumpymask
13
Chiel

ループがなければ、次のように書くことができます。

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

例えば：

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

与える：

本当

8
Bart

numpy.broadcast_to （Numpy 1.10.0の新機能）：

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