From c6daf47e7667fb804088f85b6229c8f23e5cea94 Mon Sep 17 00:00:00 2001 From: Janne Sinivirta Date: Mon, 4 Sep 2017 10:19:05 +0300 Subject: [PATCH] restore EMA plots --- analyze.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/analyze.py b/analyze.py index 4a38fcf0f..252399278 100644 --- a/analyze.py +++ b/analyze.py @@ -43,6 +43,9 @@ def get_ticker_dataframe(pair: str) -> DataFrame: } for t in sorted(data['result'], key=lambda k: k['T']) if arrow.get(t['T']) > minimum_date] dataframe = DataFrame(json_normalize(data)) + dataframe['close_30_ema'] = ta.EMA(dataframe, timeperiod=30) + dataframe['close_90_ema'] = ta.EMA(dataframe, timeperiod=90) + # calculate StochRSI stochrsi = ta.STOCHRSI(dataframe) dataframe['stochrsi'] = stochrsi['fastd'] # values between 0-100, not 0-1 @@ -114,8 +117,8 @@ def plot_dataframe(dataframe: DataFrame, pair: str) -> None: fig, (ax1, ax2, ax3) = plt.subplots(3, sharex=True) fig.suptitle(pair, fontsize=14, fontweight='bold') ax1.plot(dataframe.index.values, dataframe['close'], label='close') - # ax1.plot(dataframe.index.values, dataframe['close_30_ema'], label='EMA(60)') - # ax1.plot(dataframe.index.values, dataframe['close_90_ema'], label='EMA(120)') + ax1.plot(dataframe.index.values, dataframe['close_30_ema'], label='EMA(30)') + ax1.plot(dataframe.index.values, dataframe['close_90_ema'], label='EMA(90)') # ax1.plot(dataframe.index.values, dataframe['sell'], 'ro', label='sell') ax1.plot(dataframe.index.values, dataframe['buy'], 'bo', label='buy') ax1.legend()