python实现相关系数热力图

Python实现

import seaborn as sns
import numpy as np
import matplotlib.pyplot as plt
import random
np.random.seed(1)

data = np.random.rand(8, 8)

sns.set()
# data:数据 square:是否是正方形 vmax:最大值 vmin:最小值 robust:排除极端值影响
#sns.heatmap(data=grid, square=True, vmax=20, vmin=0, robust=True)
ax = sns.heatmap(data, center=0, annot=False, vmax=1, vmin=0, robust=True)
#plt.savefig('res.png', dpi=300)
# 标题
plt.title("test")
plt.show()

运行效果如下：