z[mask] = Z[dy[mask].astype("int"), dx[mask].astype("int")]