[英]Apply Mask Array 2d to 3d
I want to apply a mask of 2 dimensions (an NxM array) to a 3 dimensional array (a KxNxM array). 我想将2维掩模(NxM数组)应用于3维数组(KxNxM数组)。 How can I do this?
我怎样才能做到这一点?
2d = lat x lon 2d = lat x lon
3d = time x lat x lon 3d =时间x lat x lon
import numpy as np
a = np.array(
[[[ 0, 1, 2],
[ 3, 4, 5],
[ 6, 7, 8]],
[[ 9, 10, 11],
[12, 13, 14],
[15, 16, 17]],
[[18, 19, 20],
[21, 22, 23],
[24, 25, 26]]])
b = np.array(
[[ 0, 1, 0],
[ 1, 0, 1],
[ 0, 1, 1]])
c = np.ma.array(a, mask=b) # this behavior is wanted
There are quite a few different ways to choose from. 有很多不同的方法可供选择。 What you want to do is align the mask (of lower dimension) to the array that has the extra dimension: the important part is that you get the number of elements in both arrays the same, as the first example shows:
你想要做的是将掩码(较低维度)与具有额外维度的数组对齐:重要的是你得到两个数组中元素的数量相同,如第一个例子所示:
np.ma.array(a, mask=np.concatenate((b,b,b))) # shapes are (3, 3, 3) and (9, 3)
np.ma.array(a, mask=np.tile(b, (a.shape[0],1))) # same as above, just more general as it doesn't require you to specify just how many times you need to stack b.
np.ma.array(a, mask=a*b[np.newaxis,:,:]) # used broadcasting
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.