From c75524cfa0288103ec37e1538ff30472f5a83b9d Mon Sep 17 00:00:00 2001 From: Olek <45364492+yardasol@users.noreply.github.com> Date: Thu, 4 Aug 2022 16:11:11 -0500 Subject: [PATCH] simplify `from_model` Co-authored-by: Paul Romano --- openmc/deplete/microxs.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/openmc/deplete/microxs.py b/openmc/deplete/microxs.py index 29ddc7268f..c4094970f9 100644 --- a/openmc/deplete/microxs.py +++ b/openmc/deplete/microxs.py @@ -94,19 +94,16 @@ class MicroXS(DataFrame): xs[rxn].load_from_statepoint(sp) # Build the DataFrame - micro_xs = cls() - for rxn in xs: - df = xs[rxn].get_pandas_dataframe(xs_type='micro') - df.index = df['nuclide'] - df.drop(['nuclide', xs[rxn].domain_type, 'group in', 'std. dev.'], axis=1, inplace=True) - df.rename({'mean': rxn}, axis=1, inplace=True) - micro_xs = concat([micro_xs, df], axis=1) + series = {} + for rx in xs: + df = xs[rx].get_pandas_dataframe(xs_type='micro') + series[rx] = df.set_index('nuclide')['mean'] # Revert to the original tallies and materials model.tallies = original_tallies model.materials = original_materials - return cls(micro_xs) + return cls(series) @classmethod def _add_dilute_nuclides(cls, chain_file, model, dilute_initial):