根据热力图也可以看出数据表里多个特征的两两相似程度。
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
作者:曦宝