From 7099d60450dd5255c7b0462581ec1872584de4c7 Mon Sep 17 00:00:00 2001 From: "matteo.zammataro" Date: Wed, 28 Aug 2024 11:55:55 +0200 Subject: [PATCH] Small fix to MicroXS from_csv and to_csv methods --- openmc/deplete/microxs.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/openmc/deplete/microxs.py b/openmc/deplete/microxs.py index 5be9875f304..5b7919b711e 100644 --- a/openmc/deplete/microxs.py +++ b/openmc/deplete/microxs.py @@ -352,11 +352,10 @@ def from_csv(cls, csv_file, **kwargs): kwargs['float_precision'] = 'round_trip' df = pd.read_csv(csv_file, **kwargs) - df.set_index(['nuclides', 'reactions', 'groups'], inplace=True) + df.set_index(['nuclides', 'reactions'], inplace=True) nuclides = list(df.index.unique(level='nuclides')) reactions = list(df.index.unique(level='reactions')) - groups = list(df.index.unique(level='groups')) - shape = (len(nuclides), len(reactions), len(groups)) + shape = (len(nuclides), len(reactions)) data = df.values.reshape(shape) return cls(data, nuclides, reactions) @@ -377,10 +376,9 @@ def to_csv(self, *args, **kwargs): Keyword arguments passed to :meth:`pandas.DataFrame.to_csv` """ - groups = self.data.shape[2] multi_index = pd.MultiIndex.from_product( - [self.nuclides, self.reactions, range(1, groups + 1)], - names=['nuclides', 'reactions', 'groups'] + [self.nuclides, self.reactions], + names=['nuclides', 'reactions'] ) df = pd.DataFrame({'xs': self.data.flatten()}, index=multi_index) df.to_csv(*args, **kwargs)