Correlation

correlation_heatmap.py

# Set the width and height of the plot
f, ax = plt.subplots(figsize=(10, 10))

# Generate correlation matrix
corr_matrix = data.corr()

# Plot using seaborn library
sns.heatmap(corr_matrix, mask=np.zeros_like(corr_matrix,dtype=np.bool),
           square=True, ax=ax, annot=True)

plt.plot()

correlation_table.py

corr_matrix = data.corr()