import quandl
import jhtalib as jhta
import matplotlib.pyplot as plt
#quandl_data = quandl.get('BCHARTS/BITSTAMPUSD', start_date='2011-01-01', end_date='2018-11-01', order='asc', collapse='daily', returns='numpy', authtoken='YOUR_AUTH_TOKEN')
quandl_data = quandl.get('BCHARTS/BITSTAMPUSD', start_date='2011-01-01', end_date='2018-11-01', order='asc', collapse='daily', returns='numpy')
df = {'datetime': [], 'Open': [], 'High': [], 'Low': [], 'Close': [], 'Volume': []}
i = 0
while i < len(quandl_data['Close']):
df['datetime'].append(str(quandl_data['Date'][i]))
df['Open'].append(float(quandl_data['Open'][i]))
df['High'].append(float(quandl_data['High'][i]))
df['Low'].append(float(quandl_data['Low'][i]))
df['Close'].append(float(quandl_data['Close'][i]))
df['Volume'].append(int(quandl_data['Volume (BTC)'][i]))
i += 1
x = df['datetime']
plt.figure(1, figsize=(20, 10))
plt.subplot(211)
plt.title('Time / Price / Ratio')
plt.xlabel('Time')
plt.ylabel('Price')
plt.grid(True)
plt.plot(x, df['Close'], color='blue')
plt.plot(x, jhta.SMA(df, 200), color='red')
plt.legend(['Close', 'SMA 200'], loc='upper left')
plt.subplot(212)
plt.xlabel('Time')
plt.ylabel('Ratio')
plt.grid(True)
plt.plot(x, [1] * len(x))
plt.plot(x, jhta.MMR(df))
plt.legend(['SMA 200', 'MMR'], loc='upper left')
plt.show()