使用matplotlib画图自定义marker

 更新时间:2023年06月25日 09:06:16   作者:qiu_xingye  
这篇文章主要介绍了使用matplotlib画图自定义marker问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教

matplotlib画图自定义marker

在matplotlib工具箱中可以画marker的高级作图函数一共有两个,分别为plot和scatter,可以画出多种marker。

但如果需要绘制一些特殊的marker,则必须通过自的定义marker来实现。

一般有两种方式可以实现:

  • 一是在绘图中插入特殊marker的icon图片
  • 二是借助于matplotlib.path中的Path类,自定义marker的Path参数。

但无论何种方式,实现的marker都要具有一般marker的特点。

marker的特点

  • 固定大小:marker的大小只依赖于markersize关键字定义的数值,不随着绘制的图形的放大或缩小而变化;
  • 可改变方向: matplotlib默认支持的某些marker可以通过传入的参数调整marker的方向

基于marker的上述两个特点,因此不能简单地通过plot绘制特殊marker的折线或者简单地进行图片插入实现。

通过插入图片实现自定义marker

from matplotlib 
import pyplot as pltimport numpy as np
'图片插入(普通)'
N = 100
x = np.linspace(0, 10, N)
y = np.sin(x)
fig = plt.figure()
ax = fig.add_subplot(111)
ax.imshow(plt.imread('icon\\icon-test.png'), extent=(x[0],x[0]+1,y[0],y[0]+1))
ax.plot(x, y)
plt.show()

如上图所示,基本可以实现marker的插入,但大小会随figure的大小变化,而且也不能改变方向。

进一步完善代码如下:

from matplotlib import pyplot as plt
from matplotlib.offsetbox import OffsetImage, AnnotationBbox
import numpy as np
'图片插入(大小不变)'
N = 100
x = np.linspace(0, 10, N)
y = np.sin(x)
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(x, y)
img = plt.imread('icon\\icon-test.png')
im = OffsetImage(img, zoom=0.3)
ab = AnnotationBbox(im, (x[0],y[0]), xycoords='data', frameon=False)
ax.add_artist(ab)
plt.show()

此时插入的marker不再随figure大小的变化而变化,但还不能做到自定义旋转角度,如果要实现该功能,则可以通过定义旋转矩阵,将plt.imread读取的图片数据img进行旋转变换以实现对marker的角度旋转。

但这种方法仍然不同灵活,不像真正的“marker”,而且效率也不高。

下面介绍如何通过Path实现自定义marker。

通过Path实现自定义marker

from matplotlib import pyplot as plt
from matplotlib.path import Path
import numpy as np
'通过Path类自定义marker'
#定义旋转矩阵
def rot(verts, az):
    #顺时针旋转
    rad = az / 180 * np.pi
    verts = np.array(verts)
    rotMat = np.array([[np.cos(rad), -np.sin(rad)], [np.sin(rad), np.cos(rad)]])
    transVerts = verts.dot(rotMat)
    return transVerts
iconMat = np.array([[-1.414, 1.414],
			[0, 0],
			[2.828, 2.828],
			[0, 0],
			[-1.414, 1.414]])
class CustomMarker(Path):
    def __init__(self, icon, az):
         if icon == "icon": 
             verts = iconMat  
         vertices = rot(verts, az)  
         super().__init__(vertices)
N = 30
x = np.linspace(0, 10, N)
y = np.sin(x)
fig = plt.figure()
ax = fig.add_subplot(111)
ax.scatter(x, y, marker=IconMarker("icon", 20), c="red", s=1000)
plt.show()

实现效果如上图所示,就像普通的marker一样,大小固定,不随figure大小变化,而且可以调节方向(旋转)。 

实现的主要原理是

matplotlib中的plot和scatter绘图函数中的marker关键字传递的值会被matplotlib.markers中的MarkerStyle类处理,而MarkerStyle类中的set_marker是可以处理path对象的。

因此,先生成自定义的marker的Path对象的路径点参数,然后定义旋转矩阵,根据旋转角度对路径点参数进行旋转,生成指定方向的marker的Path对象的路径点参数。

剩下的marker的大小/颜色等可由MarkerStyle类或plot/scatter函数其他相关的关键字属性自动处理。

https://stackoverflow.com/questions/11487797/python-matplotlib-basemap-overlay-small-image-on-map-plot 

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • 利用Python自带PIL库扩展图片大小给图片加文字描述的方法示例

    利用Python自带PIL库扩展图片大小给图片加文字描述的方法示例

    最近的一个工程项目是讲文字添加到图像上,所以下面这篇文章主要给大家介绍了关于利用Python自带PIL库扩展图片大小给图片加文字描述的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考借鉴,下面来一起看看吧。
    2017-08-08
  • python如何派生内置不可变类型并修改实例化行为

    python如何派生内置不可变类型并修改实例化行为

    这篇文章主要为大家详细介绍了python如何派生内置不可变类型并修改实例化行为,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2018-03-03
  • python基础之停用词过滤详解

    python基础之停用词过滤详解

    这篇文章主要介绍了python基础之停用词过滤详解,文中有非常详细的代码示例,对正在学习python基础的小伙伴们有很好地帮助,需要的朋友可以参考下
    2021-04-04
  • python机器学习sklearn实现识别数字

    python机器学习sklearn实现识别数字

    本文主要介绍了python机器学习sklearn实现识别数字,主要简述如何通过sklearn模块来进行预测和学习,最后再以图表这种更加直观的方式展现出来,感兴趣的可以了解一下
    2022-03-03
  • 保留已有python安装Anaconda的方法推荐

    保留已有python安装Anaconda的方法推荐

    在安装Anaconda之前,有的已经安装过一个Python版本了,但是又不想删除这个Python版本,该怎么办呢??这篇文章主要给大家介绍了关于保留已有python安装Anaconda的几种法推荐,需要的朋友可以参考下
    2023-12-12
  • flask框架实现连接sqlite3数据库的方法分析

    flask框架实现连接sqlite3数据库的方法分析

    这篇文章主要介绍了flask框架实现连接sqlite3数据库的方法,结合实例形式分析了flask框架连接sqlite3数据库的具体操作步骤与相关实现技巧,需要的朋友可以参考下
    2018-07-07
  • Python制作钉钉加密/解密工具

    Python制作钉钉加密/解密工具

    本文给大家介绍的是使用Python实现钉钉的加密解密工具的核心代码,非常的实用,对于大家学习Python加密解密非常有帮助,希望大家能够喜欢
    2016-12-12
  • Python中文分词库jieba,pkusegwg性能准确度比较

    Python中文分词库jieba,pkusegwg性能准确度比较

    这篇文章主要介绍了Python中文分词库jieba,pkusegwg性能准确度比较,需要的朋友可以参考下
    2020-02-02
  • Python采集某网站文档并保存word格式的示例

    Python采集某网站文档并保存word格式的示例

    这篇文章主要介绍了Python采集某网站文档并保存word格式的示例,我们平常需要下载文档的时候,是不是发现,要么不能下载,要么不能复制,那么我们今天来分享一下,如何用Python将这些不给下载的文档给批量下载下来,需要的朋友可以参考下
    2023-07-07
  • 使用Python的Scrapy框架编写web爬虫的简单示例

    使用Python的Scrapy框架编写web爬虫的简单示例

    这篇文章主要介绍了使用Python的Scrapy框架编写web爬虫的简单示例,使用Python编写爬虫是Python应用方面最得意的利器,Scrapy框架正是为爬虫而生,需要的朋友可以参考下
    2015-04-04

最新评论