Merge pull request #58 from Jrenk/feature/bbands-std-deviation

feat(BBANDS): added possibility to change the factor of the standard …
peerchemist · Apr 13, 2020 · 271c9a8 · 271c9a8
2 parents 36fc4de + 64f3d90
commit 271c9a8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/finta/finta.py b/finta/finta.py
@@ -714,7 +714,7 @@ def SAR(cls, ohlc: DataFrame, af: int = 0.02, amax: int = 0.2) -> Series:
 
     @classmethod
     def BBANDS(
-        cls, ohlc: DataFrame, period: int = 20, MA: Series = None, column: str = "close"
+        cls, ohlc: DataFrame, period: int = 20, MA: Series = None, column: str = "close", std_multiplier: int = 2
     ) -> DataFrame:
         """
          Developed by John Bollinger, Bollinger Bands® are volatility bands placed above and below a moving average.
@@ -732,8 +732,8 @@ def BBANDS(
         else:
             middle_band = pd.Series(MA, name="BB_MIDDLE")
 
-        upper_bb = pd.Series(middle_band + (2 * std), name="BB_UPPER")
-        lower_bb = pd.Series(middle_band - (2 * std), name="BB_LOWER")
+        upper_bb = pd.Series(middle_band + (std_multiplier * std), name="BB_UPPER")
+        lower_bb = pd.Series(middle_band - (std_multiplier * std), name="BB_LOWER")
 
         return pd.concat([upper_bb, middle_band, lower_bb], axis=1)