matplot-热力图

最新推荐文章于 2024-09-22 16:58:21 发布

CrazyBlog

最新推荐文章于 2024-09-22 16:58:21 发布

阅读量935

点赞数

CC 4.0 BY-SA版权

分类专栏：可视化

本文链接：https://blog.csdn.net/qq_21097885/article/details/92968867

可视化专栏收录该内容

2 篇文章

订阅专栏

博客提供了两个信息技术相关参考链接，一是关于Python实现可视化热力图的内容，另一是数据可视化Seaborn颜色调控篇的教程，为数据可视化学习提供了方向。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

示例：
在这里插入图片描述
代码：

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

def plot(confusion, xticklabels, yticklabels):
	f, ax1 = plt.subplots(figsize=(10, 6),nrows=1)

	cmap = sns.light_palette("red", as_cmap=True)
	sns.heatmap(confusion, annot=True, ax=ax1, vmax=0, vmin=1, cmap=cmap)

	ax1.set_title('title')
	ax1.set_xlabel('xlabel')
	ax1.set_xticklabels(xticklabels, rotation=90) 
	ax1.set_ylabel('ylabel')
	ax1.set_yticklabels(yticklabels, rotation=0)

	f.savefig('heatmap.jpg', bbox_inches='tight')


def normalization(x):
    x_sum = np.sum(x, axis = 1, keepdims = True)
    result = x / x_sum    
    return result

if __name__ == '__main__':
	n = 10
	matrix = np.random.random((n,n))
	matrix = normalization(matrix)
	matrix = np.round(matrix, 2)

	xticklabels = []
	yticklabels = []
	for i in range(n):
		xticklabels.append('xticklabel' + str(i))
		yticklabels.append('yticklabel' + str(i))
	
	plot(matrix, xticklabels, yticklabels)

参考：
python如何实现可视化热力图 http://www.php.cn/python-tutorials-391565.html
数据可视化Seaborn从零开始学习教程（二）颜色调控篇 https://segmentfault.com/a/1190000014966210?utm_source=tag-newest