Update climdex.py

martinvonk · Jun 9, 2023 · 7704b14 · 7704b14
1 parent d358b85
commit 7704b14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spei/climdex.py b/src/spei/climdex.py
@@ -61,7 +61,7 @@ def cdd(series: Series, threshold: float = 1.0, period: str = "365D") -> Series:
 
     w = series < threshold
 
-    return w.diff().rolling(period).sum().astype(int)
+    return w.diff().rolling(period).sum().dropna().astype(int)
 
 
 def cwd(series: Series, threshold: float = 1.0, period: str = "365D") -> Series:
@@ -72,7 +72,7 @@ def cwd(series: Series, threshold: float = 1.0, period: str = "365D") -> Series:
 
     w = series >= threshold
 
-    return w.diff().rolling(period).sum().astype(int)
+    return w.diff().rolling(period).sum().dropna().astype(int)
 
 
 def prcptot(series: Series, period: str = "1Y") -> Series: