python covariance matrix

  import numpy as np import pandas as pd import matplotlib.pyplot as plt import seaborn as sns

Example 1 Manual

  stock_a = [100, 102, 104, 103, 105] stock_b = [200, 198, 202, 204, 206]
  n = len(stock_a)
  # Step 1: Calculate the means of stock_a and stock_b mean_a = sum(stock_a) / n mean_b = sum(stock_b) / n
  # Step 2: Calculate the differences from the means and their products diff_product_sum = 0
  for i in range(n): diff_a = stock_a[i] - mean_a diff_b = stock_b[i] - mean_b diff_product_sum += diff_a * diff_b
  # Step 3: Calculate the covariance covariance = diff_product_sum / n
  print(covariance)
#The positive value of 3.6 indicates that the prices of Stock A and Stock B tend to move in the same
#direction. When the price of Stock A increases, the price of Stock B also tends to increase,
#and vice versa.

Example 2 Numpy Sample Covariance

  X = [2.1, 2.5, 3.6, 4.0] Y = [8, 10, 12, 14]
  cov_matrix = np.cov(X, Y) print("Covariance matrix:\n", cov_matrix)
  cov_xy = cov_matrix[0, 1] print("Covariance between X and Y:", cov_xy)

Example 4 Pandas Sample - No Option

  data = {'X': [2.1, 2.5, 3.6, 4.0], 'Y': [8, 10, 12, 14]}
  df = pd.DataFrame(data)
  cov_df = df.cov()
  print("Covariance DataFrame:\n", cov_df)
  cov_xy = cov_df.loc['X', 'Y']
  print("Covariance between X and Y:", cov_xy)

Example 5 Plot Covariance Matrix

  # Plot the covariance matrix as a heatmap plt.figure(figsize=(8, 6)) sns.heatmap(cov_matrix, annot=True, cmap='coolwarm', fmt='.2f') plt.title('Covariance Matrix') plt.show()
  plt.scatter(X, Y) plt.xlabel('X') plt.ylabel('Y') plt.title('Scatter plot of X vs Y') plt.show()

Ryan is a Data Scientist at a fintech company, where he focuses on fraud prevention in underwriting and risk. Before that, he worked as a Data Analyst at a tax software company. He holds a degree in Electrical Engineering from UCF.

Leave a Reply

Your email address will not be published. Required fields are marked *