From 0d819a5316a354df6c024425a85e21c64ffe36dd Mon Sep 17 00:00:00 2001 From: Sander Roosendaal Date: Mon, 15 Apr 2024 21:33:37 +0200 Subject: [PATCH] interactive_plot pandas to polars --- rowers/interactiveplots.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/rowers/interactiveplots.py b/rowers/interactiveplots.py index c04dd167..a0a50b36 100644 --- a/rowers/interactiveplots.py +++ b/rowers/interactiveplots.py @@ -1375,26 +1375,24 @@ def interactive_chart(id=0, promember=0, intervaldata={}): TOOLS = 'pan,box_zoom,wheel_zoom,reset,tap,hover,crosshair' columns = ['time', 'pace', 'hr', 'fpace', 'ftime', 'spm'] - datadf = dataprep.getsmallrowdata_db(columns, ids=[id]) - - - datadf.dropna(axis=0, how='any', inplace=True) + datadf = dataprep.getsmallrowdata_pl(columns, ids=[id]) + datadf = datadf.fill_nan(None).drop_nulls() row = Workout.objects.get(id=id) - if datadf.empty: + if datadf.is_empty(): return "", "No Valid Data Available" try: _ = datadf['spm'] except KeyError: # pragma: no cover - datadf['spm'] = 0 + datadf = datadf.with_columns((pl.lit(0)).alias("spm")) try: _ = datadf['pace'] except KeyError: # pragma: no cover - datadf['pace'] = 0 + datadf = datadf.with_columns((pl.lit(0)).alias("pace")) - data_dict = datadf.to_dict("records") + data_dict = datadf.to_dicts() metrics_list = [{'name': name, 'rowingmetrics':d } for name, d in metrics.rowingmetrics]