cs660-listing#7.py

aapl_amzn_dict = { 'AAPL': log_returns.AAPL, 'AMZN': log_returns.AMZN }
AAPL_AMZN_log_returns = pd.DataFrame.from_dict(aapl_amzn_dict)

fig = plt.figure(figsize=[16,9])
ax = fig.add_subplot(2,1,2)

for c in AAPL_AMZN_log_returns:
    ax.plot(AAPL_AMZN_log_returns.index, 100*(np.exp(AAPL_AMZN_log_returns[c].cumsum()) - 1), label=str(c))

ax.set_ylabel('Total relative returns (%)')
ax.legend(loc='best')
ax.grid()

plt.show()