# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: Python 3
#     name: python3
# ---

# + id="E_Nk8PYEY7F6"
from sklearn.cluster import KMeans
import pandas as pd
import numpy as np
import pickle
import matplotlib.pyplot as plt
import seaborn as sn

# + id="XFmyOZaeZwSJ"
X = pd.read_csv("/combined_data_no-states.csv", sep=",", header = 1)
corrMatrix = X.corr()
corrMatrix.to_excel("corrMatrix.xlsx")

# + id="LmwK4bPcaYHe" outputId="d06cb81f-df77-4746-b1ab-11e61dbb3429" colab={"base_uri": "https://localhost:8080/", "height": 1000}
sn.heatmap(corrMatrix, annot=False, xticklabels=True, yticklabels = True)
figure = plt.gcf()
figure.set_size_inches(20, 20)
plt.savefig("corrMatrix", dpi=100)
plt.show()

# + id="xgsbjoRQbsd7"


StarcoderdataJupyterScriptsDedupFiltered