Tuesday, 1 June 2021

seaborn heatmap

 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