Skip to content

Commit 57a416d

Browse files
author
Dayal Chand Aichara
committed
V-1.0.0
1 parent 190cc88 commit 57a416d

File tree

1 file changed

+20
-19
lines changed

1 file changed

+20
-19
lines changed

PriceIndices/price_indicators.py

Lines changed: 20 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def get_bvol_graph(df):
6161
ax2.tick_params(axis='y', colors='b')
6262

6363
plt.suptitle('Price and Volatility Index', color='red', fontsize=24)
64-
plt.savefig('bvol_index.png',bbox_inches='tight', facecolor='orange')
64+
plt.savefig('bvol_index.png', bbox_inches='tight', facecolor='orange')
6565
return plt.show()
6666
except Exception as e:
6767
return e
@@ -107,8 +107,8 @@ def get_rsi(df):
107107

108108
data['RSI_2'] = 100 * (1 - (1 / (1 + data['RS_Smooth'])))
109109
data = data.fillna(0).reset_index(drop=True)
110-
111-
return data
110+
data1 = data
111+
return data1
112112
except Exception as e:
113113
return e
114114

@@ -173,11 +173,11 @@ def get_bollinger_bands(df, days=20):
173173
data['SD'] = data['price'].rolling(days).std()
174174
data['pluse'] = data['SMA'] + data['SD']*2
175175
data['minus'] = data['SMA'] - data['SMA']*2
176-
176+
data1 = data
177177
fig, ax = plt.subplots(figsize=(16, 12))
178-
plt.plot(data['date'], data['pluse'], color='g')
179-
plt.plot(data['date'], data['minus'], color='g')
180-
plt.plot(data['date'], data['price'], color='orange')
178+
plt.plot(data1['date'], data1['pluse'], color='g')
179+
plt.plot(data1['date'], data1['minus'], color='g')
180+
plt.plot(data1['date'], data1['price'], color='orange')
181181
plt.legend()
182182
plt.xlabel('Time', color='b', fontsize=22)
183183
plt.ylabel('Price', color='b', fontsize=22)
@@ -187,7 +187,7 @@ def get_bollinger_bands(df, days=20):
187187
plt.grid()
188188
plt.savefig('bollinger_bands.png', bbox_inches='tight', facecolor='orange')
189189
plt.show()
190-
return data
190+
return data1
191191
except Exception as e:
192192
return e
193193

@@ -213,11 +213,11 @@ def get_moving_average_convergence_divergence(df):
213213
data['EMA_12'] = data['price'].ewm(span=12, adjust=False).mean()
214214
data['EMA_26'] = data['price'].ewm(span=26, adjust=False).mean()
215215
data['MACD'] = data['EMA_12'] - data['EMA_26']
216-
data.dropna(inplace=True)
216+
data1 = data.dropna()
217217

218218
fig, ax = plt.subplots(figsize=(14, 9))
219-
plt.plot(data['date'], data['price'], color='r', label='Price')
220-
plt.plot(data['date'], data['MACD'], color='b', label='MACD')
219+
plt.plot(data1['date'], data1['price'], color='r', label='Price')
220+
plt.plot(data1['date'], data1['MACD'], color='b', label='MACD')
221221
plt.legend()
222222
plt.title('Price and MACD Plot', fontsize=28, color='b')
223223
plt.xlabel('Time', color='b', fontsize=19)
@@ -226,7 +226,7 @@ def get_moving_average_convergence_divergence(df):
226226
fig.set_facecolor('orange')
227227
plt.show()
228228

229-
return data
229+
return data1
230230
except Exception as e:
231231
return print('MACD Error - {}'.format(e))
232232

@@ -240,18 +240,18 @@ def get_simple_moving_average(df, days=15):
240240
try:
241241
data = df
242242
data['SMA'] = data['price'].rolling(days).mean()
243-
data.dropna(inplace=True)
243+
data1 = data.dropna()
244244
fig, ax = plt.subplots(figsize=(14, 9))
245-
plt.plot(data['date'], data['price'], color='r', label='Price')
246-
plt.plot(data['date'], data['SMA'], color='b', label='SMA')
245+
plt.plot(data1['date'], data1['price'], color='r', label='Price')
246+
plt.plot(data1['date'], data1['SMA'], color='b', label='SMA')
247247
plt.legend()
248248
plt.title('Price and SMA Plot', fontsize=28, color='b')
249249
plt.xlabel('Time', color='b', fontsize=19)
250250
plt.ylabel('Price', color='b', fontsize=19)
251251
plt.savefig('sma.png', bbox_inches='tight', facecolor='orange')
252252
fig.set_facecolor('orange')
253253
plt.show()
254-
return data
254+
return data1
255255
except Exception as e:
256256
return print('SMA Error - {}'.format(e))
257257

@@ -270,18 +270,19 @@ def get_exponential_moving_average(df, periods=[20]):
270270
for period in periods:
271271
data['EMA_{}'.format(period)] = data['price'].ewm(span=period, adjust=False).mean()
272272
data = data.dropna()
273+
data1 = data
273274
fig, ax = plt.subplots(figsize=(14, 9))
274-
plt.plot(data['date'], data['price'], color='r', label='Price')
275+
plt.plot(data1['date'], data1['price'], color='r', label='Price')
275276
for period in periods:
276-
plt.plot(data['date'], data['EMA_{}'.format(period)], label='EMA_{}'.format(period))
277+
plt.plot(data1['date'], data1['EMA_{}'.format(period)], label='EMA_{}'.format(period))
277278
plt.legend()
278279
plt.title('Price and EMA Plot', fontsize=28, color='b')
279280
plt.xlabel('Time', color='b', fontsize=19)
280281
plt.ylabel('Price/EMA', color='b', fontsize=19)
281282
plt.savefig('ema.png', bbox_inches='tight', facecolor='orange')
282283
fig.set_facecolor('orange')
283284
plt.show()
284-
return data
285+
return data1
285286
except Exception as e:
286287
return print('EMA Error - {}'.format(e))
287288

0 commit comments

Comments
 (0)