Maybe n_zeros should be named n_zeros_axis?