From 2a383c721a7809744016a86a4a53545ddacc5814 Mon Sep 17 00:00:00 2001 From: George Halal Date: Mon, 18 Jul 2022 15:24:19 -0700 Subject: [PATCH] Updated read_ps to work for 1 ell bin Generalized the function to work for cases in which there is only 1 broadband multipole bin --- pspy/so_spectra.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/pspy/so_spectra.py b/pspy/so_spectra.py index 5ba56a4..60f7735 100644 --- a/pspy/so_spectra.py +++ b/pspy/so_spectra.py @@ -262,11 +262,20 @@ def read_ps(file_name, spectra=None): """ data = np.loadtxt(file_name) - if spectra is None: - return data[:, 0], data[:, 1] + if data.ndim == 1: + if spectra is None: + return data[0], data[1] + + l = data[0] + ps = {spec: data[i + 1] for i, spec in enumerate(spectra)} - l = data[:, 0] - ps = {spec: data[:, i + 1] for i, spec in enumerate(spectra)} + else: + if spectra is None: + return data[:, 0], data[:, 1] + + l = data[:, 0] + ps = {spec: data[:, i + 1] for i, spec in enumerate(spectra)} + return l, ps def write_ps_hdf5(file, spec_name, l, ps, spectra=None):