seaborn heatmap matrix correlation
import seaborn as sns
import matplotlib.pyplot as plt
plt.figure(figsize=(20,8))
# taking all rows
df_small = df
correlation_mat = df_small.corr()
sns.heatmap(correlation_mat, annot = True, cmap='gist_yarg')
plt.show()
Where to get the colormap?
https://matplotlib.org/stable/tutorials/colors/colormaps.html
No comments:
Post a Comment