some improvements

2017-12-13 20:53:02 +01:00
parent 1a18f033b4
commit 8590dc7a88
4 changed files with 40 additions and 19 deletions
--- a/rowers/metrics.py
+++ b/rowers/metrics.py
@@ -343,11 +343,14 @@ def getagegrouprecord(age,sex='male',weightcategory='hwt',
            )
        )

-    ages = df['age']
-    powers = df['power']
+    if not df.empty:
+        ages = df['age']
+        powers = df['power']

-    poly_coefficients = np.polyfit(ages,powers,6)
+        poly_coefficients = np.polyfit(ages,powers,6)

-    power = np.polyval(poly_coefficients,age)
+        power = np.polyval(poly_coefficients,age)
+    else:
+        power = 0

    return power