z[z_inf] = np.nan