unzhakov/beta_shape_pnpi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: use underscores in branch names for CSV headers in concise mode

Browse Source
This commit is contained in:
unzhakov
2026-07-03 15:21:44 +03:00
parent c2795bf86e
commit c6aaf3b76a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
beta_spectrum/visualize/__init__.py
+3 -3
View File
@@ -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")