adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use TA-lib to calculate StochRSI

Browse Source
This commit is contained in:
Janne Sinivirta 2017-09-02 11:41:30 +03:00
parent c2d7854d62
commit 0643438580
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
analyze.py
View File

@ -5,6 +5,7 @@ import arrow
import requests import requests
from pandas.io.json import json_normalize from pandas.io.json import json_normalize
from stockstats import StockDataFrame from stockstats import StockDataFrame
import talib.abstract as ta
logging.basicConfig(level=logging.DEBUG, logging.basicConfig(level=logging.DEBUG,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s') format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@ -42,12 +43,9 @@ def get_ticker_dataframe(pair: str) -> StockDataFrame:
dataframe = StockDataFrame(json_normalize(data)) dataframe = StockDataFrame(json_normalize(data))
# calculate StochRSI # calculate StochRSI
window = 14 stochrsi = ta.STOCHRSI(dataframe)
rsi = dataframe['rsi_{}'.format(window)] dataframe['stochrsi'] = stochrsi['fastd'] # values between 0-100, not 0-1
rolling = rsi.rolling(window=window, center=False)
low = rolling.min()
high = rolling.max()
dataframe['stochrsi'] = (rsi - low) / (high - low)
return dataframe return dataframe
@ -59,13 +57,13 @@ def populate_trends(dataframe: StockDataFrame) -> StockDataFrame:
""" """
""" """
dataframe.loc[ dataframe.loc[
(dataframe['stochrsi'] < 0.20) (dataframe['stochrsi'] < 20)
& (dataframe['close_30_ema'] > (1 + 0.0025) * dataframe['close_60_ema']), & (dataframe['close_30_ema'] > (1 + 0.0025) * dataframe['close_60_ema']),
'underpriced' 'underpriced'
] = 1 ] = 1
""" """
dataframe.loc[ dataframe.loc[
(dataframe['stochrsi'] < 0.20) (dataframe['stochrsi'] < 20)
& (dataframe['macd'] > dataframe['macds']), & (dataframe['macd'] > dataframe['macds']),
'underpriced' 'underpriced'
] = 1 ] = 1
@ -123,8 +121,8 @@ def plot_dataframe(dataframe: StockDataFrame, pair: str) -> None:
ax2.legend() ax2.legend()
ax3.plot(dataframe.index.values, dataframe['stochrsi'], label='StochRSI') ax3.plot(dataframe.index.values, dataframe['stochrsi'], label='StochRSI')
ax3.plot(dataframe.index.values, [0.80] * len(dataframe.index.values)) ax3.plot(dataframe.index.values, [80] * len(dataframe.index.values))
ax3.plot(dataframe.index.values, [0.20] * len(dataframe.index.values)) ax3.plot(dataframe.index.values, [20] * len(dataframe.index.values))
ax3.legend() ax3.legend()
# Fine-tune figure; make subplots close to each other and hide x ticks for # Fine-tune figure; make subplots close to each other and hide x ticks for