From c6aaf3b76a23b95c6049a2d331ecbf3d5616605f Mon Sep 17 00:00:00 2001 From: Evgeniy Unzhakov Date: Fri, 3 Jul 2026 15:21:44 +0300 Subject: [PATCH] fix: use underscores in branch names for CSV headers in concise mode --- beta_spectrum/visualize/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/beta_spectrum/visualize/__init__.py b/beta_spectrum/visualize/__init__.py index 30763e2..2b32e9a 100644 --- a/beta_spectrum/visualize/__init__.py +++ b/beta_spectrum/visualize/__init__.py @@ -742,14 +742,14 @@ class BetaSpectrumAnalyzer: neutrino_mass_keV = self.config.neutrino_mass_MeV * 1000 if branch_endpoint_keV <= neutrino_mass_keV: # Branch is degenerate, spectrum should be all zeros - data[f"branch_{i + 1}_spectrum_normalized"] = np.zeros_like(branch_spectrum) + data[f"branch_{i + 1}"] = np.zeros_like(branch_spectrum) else: # Branch is valid, normalize as usual sum_spectrum = np.sum(branch_spectrum) if sum_spectrum > 0: - data[f"branch_{i + 1}_spectrum_normalized"] = branch_spectrum / sum_spectrum + data[f"branch_{i + 1}"] = branch_spectrum / sum_spectrum else: - data[f"branch_{i + 1}_spectrum_normalized"] = branch_spectrum + data[f"branch_{i + 1}"] = branch_spectrum df = pd.DataFrame(data) df.to_csv(filename, index=False, float_format="%.4e")