Matplotlib 可视化之注释与文本高级应用

最新推荐文章于 2024-03-11 13:58:14 发布

AI科技大本营

最新推荐文章于 2024-03-11 13:58:14 发布

阅读量297

点赞数

文章标签： python 数据分析可视化数据可视化深度学习

原文链接：https://mp.weixin.qq.com/s?__biz=Mzg4NDQwNTI0OQ==&mid=2247560150&idx=3&sn=874bc4e13c2755fb8f1923b4d8e39c38&chksm=cfbb0eb8f8cc87ae1977cba9dcc36dd5d6681039bad6c8fd8db6041ee368e0a09acb45e04e55&scene=126&&sessionid=0

版权

作者 | 云朵君

来源丨数据STUDIO

装饰物指定所有你可以添加到一个画布的额外元素，这样可以看美化它或使它表达更清晰。装饰物包括一些标准元素，如图例、注释、颜色条、文本等，但你也可以专门为你的画布设计自己的元素。今天和大家一起学习标准元素(注释和文本)以及特定元素(使用垂直标记在x轴上报告的根)的混合。

定义贝塞尔函数：

402 Payment Required

可以使用scipy.special.jn()函数，其中需要计算整数阶贝塞尔函数 Jn 的零点，可以使用函数 scipy.special.jn_zeros(n, nt)

计算贝塞尔函数在区间的个零点，零按升序返回。请注意，此间隔不包括在，存在于。

：贝塞尔函数的阶
：要返回的零数

绘图前准备

import numpy as np
from scipy.special import jn, jn_zeros
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['Songti SC'] # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
plt.rc("xtick", labelsize="small")
plt.rc("ytick", labelsize="small")

设置画布

fig = plt.figure(figsize=(8, 4), dpi=100)
ax = plt.subplot(xlim=[0, 20], ylim=[-0.5, 1])

绘图

i = 0
X = np.linspace(0, 20, 1000, endpoint=True)
Ji = jn(i, X)
# 配置图表样式参数
linewidth = 1.5 if i == 0 else 1
linestyle = "-" if i == 0 else "-"
color = "C1" if i == 0 else "%.2f" % (i / n)
label = r"$J_%d$" % i
# 绘图
ax.plot(
  X,
  Ji,
  color=color,
  clip_on=False,
  zorder=10 - i, # 图形叠放次序
  linewidth=linewidth,
  linestyle=linestyle,
  label=label,
)

配置图例

使用图例函数

ax.legend()

使用文字注释

k = np.argmax(Ji) # 获取曲线最大值的索引
# 在曲线最大位置上面标注
ax.text(
    X[k],
    Ji[k] + 0.05,
    label,
    color=color,
    usetex=True,  # 使用Latex语法
    ha="center",
    va="bottom",
    size="small",
  )

标注出函数等于零时的根

# 计算整数阶贝塞尔函数 Jn 的零点
Zx = [x for x in jn_zeros(i, 6) if x < 20]
# y 值恒等于0
Zy = np.zeros(len(Zx))
# 绘制散点图
ax.scatter(Zx, Zy, s=15, 
           zorder=20, 
           edgecolor=color, 
           facecolor="white", 
           linewidth=1)

# 箭头标注
ax.annotate(
    "Root",   # 标注文字
    (Zx[0], Zy[0]), 
    size="small",
    xytext=(-30, -30),
    textcoords="offset points",
    arrowprops=dict(arrowstyle="->", connectionstyle="arc3,rad=-0.3"),
  )
# y 等于0 的一条线
ax.axhline(0, color="0.5", linewidth=0.5)

标注函数公式及名称

ax.set_title(
    "Bessel functions",
    x=1,
    weight="light", ha="right",      
    family="Hiragino Sans",
    transform=ax.transAxes, 
)

ax.text(
    1, 0.98,
    r"$J_n(x) = \frac{1}{2\pi} \int_{-\pi}^\pi e^{i(x \sin \tau -n \tau)} \,d\tau$",
    va="top", ha="right",       
    transform=ax.transAxes,      
    size=12,  usetex=True,)

配置坐标轴

Zy = -0.6 * np.ones(len(Zx))
# 在x轴上用散点图的方式绘制出每个贝塞尔函数零点的刻度标识
ax.scatter(
        Zx, Zy,
        s=30, zorder=20,
        clip_on=False,
        marker="|",
        facecolor="black",
        linewidth=0.5,
      )
# 设置轴刻度标签
ax.set_yticks([-0.5, 0, 0.5, 1])
ax.set_xticks([0, 10, 20])
# 隐藏上面和右边的轴线
ax.spines["right"].set_visible(False)
ax.spines["top"].set_visible(False)
# 调整左边和底部的轴线位置
ax.spines["left"].set_position(("data", -1))
ax.spines["bottom"].set_position(("data", -0.6))

绘制多阶贝塞尔函数

细节处理

因为有多条线的交叉，我们设置白色间隙，在绘制每条函数折线时，同时绘制一条位于其下方、宽度略宽于对应折线的白色折线。因此在视觉效果上更加清晰分辨二者。

ax.plot(X, Ji, color="white", clip_on=False,
        zorder=10 - i, linewidth=2.5)

绘制过程视频欣赏

附录：报错与解决

报错:
No such file or directory: 'latex'.
解决:

windows系统下的安装步骤:

第一步：安装MiKTeX；
第二步：安装dviping(a DVI-to-PNG convert):MiKTeX中的miktex包含安装程序dviping.exe，在路径"bin/x64/dvipng.exe"下；
第三部：安装Ghostscript^[1]；安装完成后需要进行一些必要的软件安装和PATH环境变量的配置。

mac系统下的安装步骤：

第一步：安装python模块latex：pip3 install latex
第二步：安装latex环境MacTex^[2]：brew install mactex 或者直接在官网下载安装包^[3]

查看是否安装成功 latex --version或者which latex
latex的使用参考文档^[4]

第三步：MacTex安装成功后打开Tex Live Utility app然后将里面可更跟新的全部更新。参考文档^[5]

⚠️记得，如果上述办法都试过，还不行，重新电脑！