From 18907effa86e5b7d56f2b68e241b6bf789049413 Mon Sep 17 00:00:00 2001 From: Aki Date: Sat, 8 Oct 2022 11:22:36 +0200 Subject: Added helpers to access specific scalar values --- windy/point_forecast.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/windy/point_forecast.py b/windy/point_forecast.py index 7e9304f..8ee4fe2 100644 --- a/windy/point_forecast.py +++ b/windy/point_forecast.py @@ -136,6 +136,12 @@ class Prediction: def levels(self, parameter) -> tuple: return self._response.levels(parameter) + def level(self, parameter, level) -> int: + return self._response.level(parameter, level) + + def at(self, parameter, level): + return self._response.raw_predictions[parameter][self._index][self.level(parameter, level)] + def __iter__(self): return iter(self.parameters) @@ -207,6 +213,9 @@ class Response: def levels(self, parameter) -> tuple: return self._levels[parameter] + def level(self, parameter, level) -> int: + return self.levels(parameter).index(Level(level)) + def predictions(self) -> Prediction: """ Yields Prediction for each time point available in this Response. -- cgit v1.1