Matplotlib - 绘制 带有对角线的散点图 (Diagonal Scatter Plots) 函数源码

news/2024/11/22 20:12:08/

欢迎关注我的CSDN:https://spike.blog.csdn.net/
本文地址:https://blog.csdn.net/caroline_wendy/article/details/131384440

Scatter Plots

Matplotlib 是一个用于绘制二维图形的 Python 库,提供了一个 pyplot 模块,用于创建各种类型的图表。其中一种图表是散点图(Scatter Plots),用于展示两个变量之间的关系,以及数据的分布情况。要绘制散点图,使用 pyplot.scatter() 函数,接受以下参数:

  • x, y:表示数据点的横纵坐标,可以是浮点数或者数组。
  • s:表示数据点的大小,可以是一个常数或者一个数组,单位是点的平方(1 点 = 1/72 英寸)。
  • c:表示数据点的颜色,可以是一个常数、一个数组、一个颜色序列或者一个颜色字符串。如果是一个数组,那么会根据 cmap 参数来映射颜色;如果是一个颜色序列,那么长度必须和数据点相同;如果是一个颜色字符串,那么所有数据点都使用该颜色。
  • marker:表示数据点的形状,可以是一个 MarkerStyle 实例或者一个简写字符串。详细的形状列表可以参考 matplotlib.markers 文档。
  • cmap:表示用于映射颜色的 Colormap 实例或者注册的颜色图名称。如果 c 是 RGB(A) 值,那么该参数会被忽略。
  • norm:表示用于将数据标准化到 [0, 1] 区间的 Normalize 实例或者名称。默认情况下,使用线性标准化,将最小值映射到 0,最大值映射到 1。如果 c 是 RGB(A) 值,那么该参数会被忽略。
  • vmin, vmax:当使用标量数据并且没有指定 norm 时,vmin 和 vmax 定义了颜色图覆盖的数据范围。默认情况下,颜色图覆盖了所有数据的值域。

主要:

  1. 设置主题:sns.set_theme(style="white")
  2. 设置 Figure 尺寸:fig.set_size_inches(10, 10)
  3. 绘制对角线:ax.plot()
  4. 绘制散点图:ax.scatter()
  5. 绘制标注:ax.annotate()

源码如下:

#!/usr/bin/env python
# -- coding: utf-8 --
"""
Copyright (c) 2022. All rights reserved.
Created by C. L. Wang on 2023/6/25
"""
import osimport seaborn as snssns.set_theme(style="white")import matplotlib.pyplot as plt
import numpy as npfrom myutils.project_utils import read_excel_to_df
from root_dir import DATA_DIRdef draw_diagonal_scatter_plots(data_better, data_ref, label_list,min_scale=0.0, max_scale=1.05,x_label="", y_label="",save_name=""
):"""绘制对角线散点图:param data_better: 优质数据,数据位于右下方:param data_ref:    对比数据:param label_list:  标签:param min_scale:   最小范围:param max_scale:   最大范围:param x_label:     x轴描述:param y_label:     y轴描述:param save_name:   文件存储:return: 图"""assert len(data_ref) == len(data_better) == len(label_list)fig, ax = plt.subplots()fig.set_size_inches(10, 10)ax.grid(True)ax.plot([min_scale, max_scale], [min_scale, max_scale], ls="--", c=".3")ax.scatter(data_ref, data_better, s=100, edgecolors="black")plt.xlim(min_scale, max_scale)plt.ylim(min_scale, max_scale)plt.xticks(fontsize=15)plt.yticks(fontsize=15)plt.xlabel(x_label, fontsize=20)plt.ylabel(y_label, fontsize=20)for i, txt in enumerate(label_list):if data_ref[i] - data_better[i] > 0.025:ax.annotate(txt, (data_ref[i] * 1.03, data_better[i]), fontsize=10, fontweight='bold')if save_name:# transparent=Trueassert save_name.endswith("png") or save_name.endswith("jpg")plt.savefig(save_name, bbox_inches='tight', format='png')plt.show()def main():df = read_excel_to_df(os.path.join(DATA_DIR, "Strategy-v1v2v3v4-TMScore.xls"))data1 = df["m0-score"]data2 = df["max-score"]label_list = df["target"]print(f"data1 : {round(float(np.mean(data1)) * 100, 4)}±{round(float(np.std(data1)), 4)}")print(f"data2 : {round(float(np.mean(data2)) * 100, 4)}±{round(float(np.std(data2)), 4)}")draw_diagonal_scatter_plots(data1, data2, label_list, min_scale=0.35, max_scale=1.05,x_label="Our P.S.P. (TMScore)", y_label="CASP15 SOTA (TMScore)",save_name="img.png")if __name__ == '__main__':main()

http://www.ppmy.cn/news/571000.html

相关文章

蓝牙无线耳机哪个品牌好?盘点前七名无线蓝牙耳机分享

本文将为您详细介绍每款蓝牙耳机的设计特点、音质表现、续航能力和智能功能等关键信息。我们将提供客观、全面的分析,帮助您更好地了解每款产品的优势和适用场景;无论您是追求高保真音质的音乐发烧友,还是需要轻便舒适的耳机进行运动&#xf…

我做了10年的测试,由衷的建议年轻人别入这行了

两天前,有个做功能测试7年的同事被裁员了。这位老哥已经做到了团队中的骨干了,人又踏实,结果没想到刚刚踏入互联网“老龄化”大关,就被公司给无情优化了。 现在他想找同类型的工作,薪资也一直被压,考虑转行…

小白组装电脑(上手阶段)

11月1号凌晨下的单,2号货就全部到齐了,狗东服务就是好。 第一天 把所有的配件理了一遍, 明白每个配件的对应主板接口、电源接口以及要用到哪些线。 买机械盘都没有配任何线,反而是主板配了两条用来接硬盘的SATA线。固态买的M.2…

主流低功耗服务器u,新组低功耗NAS服务器(1037U)分享

" ]5 ?6 B/ P6 \; v) i [1 u此贴作为经验分享和交流,参考个人应用需求,简要论述NAS选型,供各位参考! , m8 q; M# Y* D% M0 Q; u9 T 3 X2 S& u$ W- S) F9 }! D; E 9 A4 K; m8 f$ {. u7 q ! C7 J0 `8 y: ?新组建NAS原因和需求 " w: ^0 p8 W" a9 P% E# W…

tcpdump参数使用举例

tcpdump 是一款强大的网络抓包工具,它可以对网络流量进行捕获和分析。下面是 tcpdump 常用的一些参数及其使用方法。 -i 参数 -i 参数用于指定要捕获的网络接口。例如,-i eth0 表示捕获 eth0 接口的网络流量。如果不指定该参数,则 tcpdump …

Python+requests+unittest+excel搭建接口自动化测试框架

一、框架结构: 工程目录 代码:基于python2编写 二、Case文件设计 三、基础包 base 3.1 封装get/post请求(runmethon.py) import requests import json class RunMethod:def post_main(self,url,data,headerNone):res Noneif h…

基于企业微电网平台的食用菌工厂的能源管理

摘要:从食用菌生产经营者角度指出食用菌工厂化生产的产品生鲜属性、微生物培养的技术特点和竞争战略难以突破决定了成本控制是经营的核心。分析食用菌生产的成本构成,指出加强能源管理是节本增效的重要途径,从电费的分类和改善方式、灭菌的燃…

深度学习(26)——YOLO系列(5)

深度学习(26)——YOLO-v7(5) 文章目录 深度学习(26)——YOLO-v7(5)絮絮叨叨1. conv和BN的融合2. 3*3卷积的替换(1)1*1卷积有什么作用?&#xff08…