diff --git a/assignments/a3/a3_part4.py b/assignments/a3/a3_part4.py index d7d50d7..3f55089 100644 --- a/assignments/a3/a3_part4.py +++ b/assignments/a3/a3_part4.py @@ -74,6 +74,16 @@ def get_xy_data(outputs: dict[tuple[int, int], FfwiOutput], attribute: str) -> \ >>> getattr(output, 'ffmc') 2.0 """ + # Accumulators + days: list[str] = [] + data: list[float] = [] + + for k in outputs.keys(): + month, day = k + days.append(f'{month}, {day}') + data.append(getattr(outputs[k], attribute)) + + return days, data def plot_ffwi_attribute(outputs: dict[tuple[int, int], FfwiOutput], attribute: str) -> None: