matplotlib 绘制椭圆以注解打印

hc8w905p  于 2023-10-24  发布在  其他
关注(0)|答案(2)|浏览(98)

我最近发现matplotlib是一个比Matlab更好的绘图替代品。不幸的是,我对python的了解接近于零。
考虑下图和相关代码。

import numpy as np
import scipy.io
from matplotlib import pyplot as plt

plt.figure(figsize=[3.3, 3.3])
plt.rcParams.update({'font.size': 8, 'text.usetex': True})

plt.plot([1, 2, 3, 4], [1, 4, 9, 16], color='r')
plt.plot([1, 2, 3, 4], [2, 5, 10, 17], color='r')
plt.plot([1, 2, 3, 4], [11, 14, 19, 26], color='b')
plt.plot([1, 2, 3, 4], [12, 15, 20, 27], color='b')

plt.annotate('blue curves', xy=(1.5, 22.5), xytext=(1.5, 22.5), ha='center', va='center')
plt.annotate('', xy=(2, 15), xytext=(1.5, 22), arrowprops=dict(width=0.1, headwidth=2, headlength=2, color='grey'))
plt.annotate('red curves', xy=(2.5, 22.5), xytext=(2.5, 22.5), ha='center', va='center')
plt.annotate('', xy=(3, 10), xytext=(2.5, 22), arrowprops=dict(width=0.1, headwidth=2, headlength=2, color='grey'))

plt.grid()
plt.xlabel('x')
plt.ylabel('y')

plt.savefig('filename.pdf', format='pdf')

plt.show()

我想在每个箭头上添加一个小椭圆/圆来包围曲线。椭圆/圆应该具有与箭头相同的风格(即颜色,厚度等)。有没有一种简单的方法来做到这一点,而无需修改现有的代码?
我试图从this example中获得一些灵感,但它没有工作。

5lhxktic

5lhxktic1#

一个简单的解决方案,如果一个圆是好的,是在箭头的位置添加一个分散点(我没有安装LaTex):

import numpy as np
import scipy.io
from matplotlib import pyplot as plt

plt.figure(figsize=[3.3, 3.3], dpi=100)
plt.rcParams.update({'font.size': 8, 'text.usetex': False})

plt.plot([1, 2, 3, 4], [1, 4, 9, 16], color='r')
plt.plot([1, 2, 3, 4], [2, 5, 10, 17], color='r')
plt.plot([1, 2, 3, 4], [11, 14, 19, 26], color='b')
plt.plot([1, 2, 3, 4], [12, 15, 20, 27], color='b')

plt.annotate('blue curves', xy=(1.5, 22.5), xytext=(1.5, 22.5), ha='center', va='center')
plt.annotate('', xy=(2, 15), xytext=(1.5, 22), arrowprops=dict(width=0.1, headwidth=2, headlength=2, color='grey'))
plt.annotate('red curves', xy=(2.5, 22.5), xytext=(2.5, 22.5), ha='center', va='center')
plt.annotate('', xy=(3, 10), xytext=(2.5, 22), arrowprops=dict(width=0.1, headwidth=2, headlength=2, color='grey'))

plt.plot(2, 15, 'o', ms=10 * 2, mec='grey', mfc='none', mew=1.5)
plt.plot(3, 10, 'o', ms=10 * 2, mec='grey', mfc='none', mew=1.5)

plt.grid()
plt.xlabel('x')
plt.ylabel('y')

plt.savefig('filename.pdf', format='pdf')

plt.show()

yhxst69z

yhxst69z2#

虽然你可以在你的plt-对象上调用plotannotate,但你需要访问figureaxis。这里给出了一个很好的解释图,图和轴之间的区别。
您需要添加的代码(在plt.show()之前):

from matplotlib.patches import Ellipse

fig = plt.gcf()  # access the current figure
ax = plt.gca()   # access the current axis
xmin, xmax = ax.get_xlim()  # get the minimum and maximum extent of the x-axis
ymin, ymax = ax.get_ylim()  # get the minimum and maximum extent of the y-axis
aspect_ratio = fig.get_figheight() / fig.get_figwidth()  # get aspect ratio, we will use it to scale the width and height of the ellipse to appear as a circle
ax.add_patch(Ellipse(xy=(2, 15), width=(xmax-xmin)*aspect_ratio/5, height=(ymax-ymin)/5, color="grey", fill=False, lw=2))
ax.add_patch(Ellipse(xy=(3, 10), width=(xmax-xmin)*aspect_ratio/5, height=(ymax-ymin)/5, color="grey", fill=False, lw=2))

相关问题