def _create_phases(self):
"""
Create a dictionary of phases.
Returns:
(covsirphy.PhaseSeries)
"""
start_dates, end_dates = self._phase_range(self._change_dates)
pop_list = [self.pop_dict[date] for date in start_dates]
phase_series = PhaseSeries(
self.dates[0], self.dates[-1], self.population, use_0th=self._use_0th
)
phase_itr = enumerate(zip(start_dates, end_dates, pop_list))
for (i, (start_date, end_date, population)) in phase_itr:
phase_series.add(
start_date=start_date,
end_date=end_date,
population=population
)
return phase_series
def _create_phases(self):
"""
Create a dictionary of phases.
Returns:
(covsirphy.PhaseSeries)
"""
start_dates, end_dates = self._phase_range(self.change_dates)
pop_list = [self.pop_dict[date] for date in start_dates]
phases = [self.num2str(num) for num in range(len(start_dates))]
phase_series = PhaseSeries(
self.dates[0], self.dates[-1], self.population
)
phase_itr = enumerate(zip(start_dates, end_dates, pop_list, phases))
for (i, (start_date, end_date, population, phase)) in phase_itr:
if i == 0:
continue
phase_series.add(
start_date=start_date,
end_date=end_date,
population=population
)
return phase_series