蒙版时数组的尺寸损失

查看

numpy.where

http://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html

为了保持相同的尺寸，您将需要一个填充值。在下面的示例中，我使用0，但您也可以使用

np.nan

np.where(arr>3, arr, 0)

退货

array([[[[ 0, 11],         [ 0, 22],         [ 0, 33]],        [[ 4, 44],         [ 5, 55],         [ 6, 66]]],       [[[ 7, 77],         [ 8, 88],         [ 9, 99]],        [[ 0, 32],         [ 0, 33],         [ 0, 34]]]])