moshferatu/strategies
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Import where() directly instead of importing the entire numpy package in RSI 25 / 75 strategy

Browse Source
This commit is contained in:
moshferatu 2025-01-07 09:29:20 -08:00
parent c24554a725
commit 21cacfaf0c
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
strategies/rsi_25_75.py
View File

@ -1,5 +1,4 @@
import numpy as np from numpy import where
from pandas import DataFrame, Series from pandas import DataFrame, Series
from indicators import sma, rsi from indicators import sma, rsi
@ -17,4 +16,4 @@ def rsi_25_75(data: DataFrame) -> Series:
rsi_below_25 = rsi_4 < 25 rsi_below_25 = rsi_4 < 25
conditions = above_ma_200 & rsi_below_25 conditions = above_ma_200 & rsi_below_25
return Series(np.where(conditions, 'L', 'N'), index = data.index) return Series(where(conditions, 'L', 'N'), index = data.index)