It's basically the same, but exchange the 0 with False here, as the mask is boolean.