tardis.plasma.equilibrium.rates.util module
- tardis.plasma.equilibrium.rates.util.reindex_ion_population_to_level_population(ion_population: DataFrame, level_population: DataFrame, next_higher: bool = True) DataFrame[source]
Align ion population indexes with level populations by ionization stage.
- Parameters:
- ion_populationpd.DataFrame
Ion-resolved values indexed by atomic number and ion number.
- level_populationpd.DataFrame
Level-resolved values indexed by atomic number, ion number, and level number.
- next_higherbool, optional
Select the continuum ion for each level when
True; otherwise select the ion containing the level.
- Returns:
- pd.DataFrame
Ion-resolved values with the level-population index and columns.
- tardis.plasma.equilibrium.rates.util.reindex_ionization_rate_dataframe(rate_dataframe: DataFrame, recombination=False)[source]
Add ion source and destination numbers the ionization rate dataframe.
- Parameters:
- rate_dataframepd.DataFrame
Dataframe of ionization rates, indexed by atomic number, ion number, level source number and level destination number.
- recombinationbool, optional
If the rates are recombination rates, by default False
- Returns:
- pd.DataFrame
Dataframe with additional columns for ion source and destination