python 格子图

滴滴答答11

于 2020-09-19 20:10:10 发布

阅读量4.2k

点赞数 2

文章标签： python

本文链接：https://blog.csdn.net/weixin_46604505/article/details/108685280

版权

import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
from scipy.stats import pearsonr, spearmanr

data1 = [[1,0.196699884,-0.077219634],[0.196699884,1,-0.668343535],[-0.077219634,-0.668343535,1]]
fig, ax = plt.subplots(figsize=(10, 10))
key_list = ['A', 'B', 'C']
sns.heatmap(pd.DataFrame(np.round(data1, 2), columns=key_list, index=key_list), annot=True, vmax=1, vmin=-1,
            square=True, cmap="YlGnBu")
ax.set_title('', fontsize=18)

效果图