python绘制热力图,数据来源pandas.dataframe

news2024/9/20 6:09:01

通过pandas.dataframe绘制热力图,并标出颜色最深即z轴数据最大的点。

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

class Heatmap:
    def __init__(self, data, marker='*', marker_color='white', marker_size=10):
        self.data = data
        self.size = len(data)
        self.marker = marker
        self.marker_color = marker_color
        self.marker_size = marker_size
        self.title = None
        self.xlabel = None
        self.ylabel = None
        self.cbar_label = None
        self.cbar_location = "right"

    def plot_heatmap(self):
        # 创建一个新的图像
        fig, ax = plt.subplots()

        # 使用imshow创建热力图
        cax = ax.imshow(self.data, cmap='turbo', interpolation='nearest')

        # 找到最大值的坐标并绘制标记
        max_pos = self._get_max_pos()
        self._plot_marker(ax, max_pos)

        # 设置刻度
        self.set_ticks(ax)

        # 设置标题和x,y轴标签
        ax.set_title(self.title)
        ax.set_xlabel(self.xlabel)
        ax.set_ylabel(self.ylabel)

        # 设置colorbar
        cbar = plt.colorbar(cax, ax=ax, label=self.cbar_label, pad=0.01, fraction=0.046)
        cbar.ax.yaxis.set_label_position(self.cbar_location)  # 设置colorbar的标签位置

        # 显示图像
        plt.show()

    def _get_max_pos(self):
        # 找到最大值的坐标
        max_value = self.data.values.max()
        max_pos = np.unravel_index(self.data.values.argmax(), self.data.values.shape)
        return max_pos

    def _plot_marker(self, ax, position):
        # 在指定位置绘制标记
        ax.plot(position[1], position[0], marker=self.marker, color=self.marker_color, markersize=self.marker_size)

    def set_labels(self, title=None, xlabel=None, ylabel=None):
        # 设置标题和x,y轴标签
        self.title = title
        self.xlabel = xlabel
        self.ylabel = ylabel

    def set_ticks(self, ax):
        # 动态设置x轴和y轴的刻度
        if self.size <= 10:
            interval = 1
        else:
            interval = self.size // 10

        ticks = np.arange(0, len(self.data.columns), interval)
        ax.set_xticks(ticks)
        ax.set_yticks(ticks)
        ax.set_xticklabels(self.data.columns[ticks])
        ax.set_yticklabels(self.data.index[ticks])

    def set_colorbar(self, label=None, location="right"):
        # 设置colorbar的标签和位置
        self.cbar_label = label
        self.cbar_location = location


# 创建一个随机数据 DataFrame,使用你所要求的行名和列名
index = range(-20, 21)
columns = range(-20, 21)
data = pd.DataFrame(np.random.rand(41, 41), index=index, columns=columns)

# 创建 Heatmap 对象
heatmap = Heatmap(data)

# 设置热力图标题和 x, y 轴标签
heatmap.set_labels('Heatmap Title', 'X-axis Label', 'Y-axis Label')

# 设置 colorbar 的标签和位置
heatmap.set_colorbar('Colorbar Label', 'right')

# 创建并显示热力图
heatmap.plot_heatmap()


在这里插入图片描述

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/709316.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

排序的概念,插入排序,希尔排序,选择排序

文章目录 排序的概念排序的概念&#xff1a;排序运用常见的排序与时间复杂度 插入排序希尔排序希尔排序的概念希尔排序总结 选择排序概念 排序的概念 排序的概念&#xff1a; 排序&#xff1a;所谓排序&#xff0c;就是使一串记录&#xff0c;按照其中的某个或某些关键字的大小…

【复习《剑指Offer》13-17题】【每天40分钟,我们一起用50天刷完 (剑指Offer)】第十三天 7/50

专注 效率 记忆 预习 笔记 复习 做题 欢迎观看我的博客&#xff0c;如有问题交流&#xff0c;欢迎评论区留言&#xff0c;一定尽快回复&#xff01;&#xff08;大家可以去看我的专栏&#xff0c;是所有文章的目录&#xff09;   文章字体风格&#xff1a; 红色文字表示&#…

什么是精益创业?精益创业的核心方法有哪些?

《经理人参阅&#xff1a;创业管理学》是一本非常有价值的创业类书籍&#xff0c;为创业者提供了许多实用的方法和经验&#xff0c;特别是关于精益创业方面的内容。精益创业是一种以快速验证假设为核心的创业方法&#xff0c;它可以帮助创业者在最短的时间内找到最有效的商业模…

Kafka跨集群备份解决方案MirrorMaker

文章目录 1 什么是 MirrorMaker&#xff1f;2 运行 MirrorMakerMirrorMaker 配置实例第 1 步&#xff1a;启动两套 Kafka 集群第 2 步&#xff1a;启动 MirrorMaker 工具第 3 步&#xff1a;验证消息是否拷贝成功 3 其他跨集群镜像方案4 小结 一般情况下&#xff0c;我们会使用…

Win2008下使用IIS+URL重写+Server Farms在同一台服务器实现混合 多域名网站与多个负载平衡集群网站 共存,配置及域名绑定笔记

因公司要在现有云服务器增加负载平衡集群网站&#xff0c;但要保证现有在用网站不动&#xff0c;操作系统为Win2008。 以前没有配置过IIS的负载平衡&#xff0c;只能百度一下&#xff0c;实现也简单&#xff1a;只要安装URL重写Server Farms就能实现负载平衡集群网站。于是根据…

IMX6ULL系统移植篇-uboot的烧写与运行

一. uboot 配置与编译 本文来学习 uboot的烧写&#xff0c;承接之前 uboot的配置与编译篇&#xff0c;博文如下&#xff1a; IMX6ULL系统移植篇-uboot的配置与编译_凌肖战的博客-CSDN博客 二. uboot的烧写 uboot 编程成功后&#xff0c;会在 uboot的根目录下生成 u-boot.b…

CentOS Linux的最佳替代方案(三)_银河麒麟Kylin Linux Advanced Server V10 SP2 for X86安装教程

文章目录 CentOS Linux的最佳替代方案&#xff08;三&#xff09;_银河麒麟Kylin Linux Advanced Server V10 SP2 for X86安装教程1 银河麒麟Kylin Linux Advanced Server V10简介1.1 同源构建1.2 自主CPU平台深入优化1.3 虚拟化及云原生支持1.4 高可用性1.5 可管理性1.6 内生本…

java阻塞队列/kafka

queue增加删除元素 增加元素 add方法在添加元素的时候&#xff0c;若超出了度列的长度会直接抛出异常&#xff1a;put方法&#xff0c;若向队尾添加元素的时候发现队列已经满了会发生阻塞一直等待空间&#xff0c;以加入元素offer方法在添加元素时&#xff0c;如果发现队列已满…

PCL点云处理之沿着法向量投影点云到平面 (一百九十四)

即使你很爱她,也要保留一些尊严 PCL点云处理之沿着法向量投影点云到平面 (一百九十四) 一、算法介绍二、具体实现1.代码(详细注释)2.效果一、算法介绍 实现这样的目标:沿着法向量,将点云投影到一个平面上 具体步骤: 1计算整个点云的法向量,来确定平面方向 2选择某一点…

IP编址数据转发

目录 一、IP编址 1.1、二进制、十进制和十六进制 1.2、进制之间的转换 1.3、IP编址 1.4、子网掩码 1.5、二进制和十进制转换 1.6、IP地址分类 1.7、IP地址类型 1.8、地址规划 二、VLSM与CIDR 2.1、有类IP编址的缺陷 2.2、变长子网掩码 VLSM 2.3、缺省情况下的掩码 …

红色通信史(四):永不消逝的电波

年龄稍大一点的读者&#xff0c;相信对下面这张剧照并不会感到陌生。 没错&#xff0c;这张经典的剧照&#xff0c;来自曾经家喻户晓的一部红色电影——《永不消逝的电波》。 这部电影上映于1958年&#xff0c;由八一电影制片厂摄制&#xff0c;王苹执导&#xff0c;孙道临、袁…

Qt和MySQL的连接

具体视频的教程 视频教程 我的文章是做一下补充的 流程&#xff1a; 第一点你要确保你的数据库的位数&#xff0c;如果你数据库位数的是32位&#xff0c;mingw编译的时候就需要选择的是32位**&#xff08;在下面会提到&#xff09;** 去到所在的文件 点击.pro文件进行编译…

ModaHub魔搭社区:为什么选择Zilliz Cloud?

目录 Zilliz Cloud是什么&#xff1f; 为什么选择Zilliz Cloud&#xff1f; 基于Milvus构建&#xff0c;针对性能进行优化 弹性和可扩展 按需付费 多云支持&#xff08;AWS、GCP&#xff09; 云原生的可靠性 企业安全和治理 Zilliz Cloud基于Milvus的云原生服务 Zilliz…

深蓝学院C++基础与深度解析笔记 第 8 章 动态内存管理

第 8 章 动态内存管理 1. 动态内存基础 ● 栈内存 V.S. 堆内存 – 栈内存的特点&#xff1a;更好的局部性&#xff0c;用于语言的固有类型&#xff0c;对象自动销毁,由低到高开辟 – 堆内存的特点&#xff1a;运行期动态扩展&#xff0c;需要显式释放&#xff0c;由高到低开辟…

使用 TailwindCSS 中的 color-mix() 构建自定义调色板

在这篇文章中&#xff0c;我们将了解如何使用 CSS 函数color-mix()和 CSS 变量&#xff0c;通过 TailwindCSS 高效地为 Nuxt 应用程序生成自定义调色板。 先决条件 最好使用以下命令设置 Nuxt 应用程序&#xff1a; npx nuxi init tailwindcss-color-mix 在安装提示期间选择 …

新项目即将启动!小灰做个市场调研

熟悉小灰的小伙伴们都知道&#xff0c;在2019年初&#xff0c;做了整整10年程序员的小灰离开职场&#xff0c;成为了一名自由职业者。 2021年末&#xff0c;小灰注册了自己的公司&#xff0c;名为北京小灰大黄科技有限公司。 公司虽然注册了&#xff0c;但是整个公司只有小灰一…

【C2】文件,时间,多线程,动静态库

文章目录 1.文件&#xff1a;fprint/fgets/fwrite/fread&#xff0c;ftell/rewind/fseek/fflush1.1 文本文件&#xff1a;FILE结构体1.2 二进制文件&#xff1a;没有行概念1.3 文件定位&#xff1a;linux下文本文件模式和二进制文件模式没有区别。fgets和fprintf以行方式读写文…

【Flutter】Flutter 国际化入门 使用 intl 包 格式化日期

文章目录 一、 前言二、 版本信息三、 什么是 intl 包四、 如何安装和使用 intl 包1. 安装 intl 包2. 使用 intl 包进行基本的日期和数字格式化3. 使用 intl 包进行消息翻译 五、 一个简单的使用示例六、 总结 一、 前言 在全球化的今天&#xff0c;为你的 Flutter 应用添加国…

快速上手MATLAB图像处理:100种项目全覆盖

本教程涵盖了MATLAB图像处理的广泛内容。我们学习了图像读取、显示和保存,图像的基本操作(如缩放、裁剪、旋转和翻转),以及图像的基本增强(如亮度调整、对比度调整和颜色空间转换)。本教程还介绍了常见的图像滤波技术(如均值滤波、中值滤波和高斯滤波),图像的直方图均…

JAVA临时文件的使用

目录 什么是临时文件&#xff1f; 临时文件在编程中有各种妙用 java在缓存目录创建临时文件的方式 1 按照指定文件名随机数字共同作为文件名创建 2 按照指定文件名创建 3 通过获取临时文件夹的真实路径 什么是临时文件&#xff1f; 临时文件是在计算机系统中用于临时存储数…