根据热力图也可以看出数据表里多个特征的两两相似程度。

image.png

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

tps = read_csv('E:\workfile\data\trade\tps.csv', header=0)
label = np.array(tps)[:, 0]
feature = np.array(tps)[:, 1:]

下面是画热力图的方法

def relitu():
dfData = tps.corr()
plt.subplots(figsize=(9, 9))  # 设置画面大小
sns.heatmap(dfData, annot=True, vmax=1, square=True, cmap="Blues")
plt.savefig('./BluesStateRelation.png')
plt.show()

image.png

作者：曦宝
链接：https://www.jianshu.com/p/2fd77b102229