diff options
Diffstat (limited to 'fg21sim/extragalactic/clusters/main.py')
| -rw-r--r-- | fg21sim/extragalactic/clusters/main.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/fg21sim/extragalactic/clusters/main.py b/fg21sim/extragalactic/clusters/main.py index 8fdefb0..074758a 100644 --- a/fg21sim/extragalactic/clusters/main.py +++ b/fg21sim/extragalactic/clusters/main.py @@ -532,6 +532,13 @@ class GalaxyClusters: os.rename(outfile, outfile+".old") logger.info("Converting halos data into a Pandas DataFrame ...") + + # Pad the merger events to be same length + nmax = max([d["merger_num"] for d in self.halos]) + padkeys = ["merger_mass1", "merger_mass2", "merger_z", "merger_t"] + for d in self.halos: + pad_dict_list(d, padkeys, length=nmax) + keys_ignored = ["gamma", "spectrum", "spectrum_fiducial", "template"] keys = [k for k in self.halos[0].keys() if k not in keys_ignored] halos_df = dictlist_to_dataframe(self.halos, keys=keys) |