@@ -90,7 +90,7 @@ def get_rsi(df):
9090 try :
9191 data = df
9292 data = data .sort_values (by = 'date' ).reset_index (drop = True )
93- data ['price_change' ] = data ['price' ] - df ['price' ].shift (1 )
93+ data ['price_change' ] = data ['price' ] - data ['price' ].shift (1 )
9494 data .dropna (inplace = True )
9595 data ['gain' ] = data ['price_change' ].apply (lambda x : x if x >= 0 else 0 )
9696
@@ -100,11 +100,11 @@ def get_rsi(df):
100100
101101 data ['loss_average' ] = data ['loss' ].rolling (14 ).mean ()
102102
103- data ['RS' ] = data ['gain_average' ] / df ['loss_average' ]
103+ data ['RS' ] = data ['gain_average' ] / data ['loss_average' ]
104104
105- data ['RSI_1' ] = 100 * (1 - (1 / (1 + df ['RS' ])))
105+ data ['RSI_1' ] = 100 * (1 - (1 / (1 + data ['RS' ])))
106106
107- data ['RS_Smooth' ] = (data ['gain_average' ].shift (1 ) * 13 + df ['gain' ]) / (
107+ data ['RS_Smooth' ] = (data ['gain_average' ].shift (1 ) * 13 + data ['gain' ]) / (
108108 data ['loss_average' ].shift (1 ) * 13 + data ['loss' ])
109109
110110 data ['RSI_2' ] = 100 * (1 - (1 / (1 + data ['RS_Smooth' ])))
@@ -175,11 +175,11 @@ def get_bollinger_bands(df, days=20):
175175 data = data .sort_values (by = 'date' ).reset_index (drop = True )
176176 data ['SMA' ] = data ['price' ].rolling (days ).mean ()
177177 data ['SD' ] = data ['price' ].rolling (days ).std ()
178- data ['pluse ' ] = data ['SMA' ] + data ['SD' ]* 2
178+ data ['plus ' ] = data ['SMA' ] + data ['SD' ]* 2
179179 data ['minus' ] = data ['SMA' ] - data ['SMA' ]* 2
180180 data1 = data .sort_values (by = 'date' , ascending = False ).reset_index (drop = True )
181181 fig , ax = plt .subplots (figsize = (16 , 12 ))
182- plt .plot (data1 ['date' ], data1 ['pluse ' ], color = 'g' )
182+ plt .plot (data1 ['date' ], data1 ['plus ' ], color = 'g' )
183183 plt .plot (data1 ['date' ], data1 ['minus' ], color = 'g' )
184184 plt .plot (data1 ['date' ], data1 ['price' ], color = 'orange' )
185185 plt .legend ()
0 commit comments