java 热力图,热力图

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

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