1. Learn
    2. /
  2. Courses
    3. /
  3. Manipulating Time Series Data in Python

Exercise

Annual return correlations among several stocks

You have seen in the video how to calculate correlations, and visualize the result.

In this exercise, we have provided you with the historical stock prices for Apple (AAPL), Amazon (AMZN), IBM (IBM), WalMart (WMT), and Exxon Mobile (XOM) for the last 4,000 trading days from July 2001 until the end of May 2017.

You'll calculate the year-end returns, the pairwise correlations among all stocks, and visualize the result as an annotated heatmap.

Instructions

100 XP

We have already imported pandas as pd, seaborn as sns, and matplotlib.pyplot as plt. We have loaded the daily close price for the five stocks in a variable called data.

  • Inspect using .info().
  • Apply .resample() with year-end frequency (alias: 'A') to data and select the .last() price from each subperiod; assign this to annual_prices.
  • Calculate annual_returns by applying .pct_change() to annual_prices.
  • Calculate correlations by applying .corr() to annual_returns and print the result.
  • Visualize correlations as an annotated sns.heatmap().