Matplotlib:绘制椭圆以在对数刻度上进行注释
Matplotlib: draw ellipse to annotate on top of logarithmic scale
我想使用 matplotlib 在半曲线图上画一个椭圆。
考虑下图和相关代码。
import numpy as np
import scipy.io
from matplotlib import pyplot as plt
from matplotlib.patches import Ellipse
plt.figure(figsize=[3.3, 3.3])
plt.rcParams.update({'font.size': 8, 'text.usetex': True})
plt.semilogy([1, 2, 3, 4], [4, 8, 12, 16], color='r')
plt.semilogy([1, 2, 3, 4], [2, 4, 6, 8], color='r')
plt.semilogy([1, 2, 3, 4], [12, 15, 20, 27], color='b')
ax = plt.gca()
plt.annotate('blue curve', 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, 15), xytext=(2.5, 22), arrowprops=dict(width=0.1, headwidth=2, headlength=2, color='grey'))
ax.add_patch(Ellipse(xy=(3, 10), width=0.2, height=10, color="grey", fill=False, lw=1, zorder=5))
plt.grid()
plt.xlabel('x')
plt.ylabel('y')
plt.savefig('filename.pdf', format='pdf')
plt.show()
如您所见,由于 y 轴上的对数刻度,椭圆变形了。我试图在自然尺度上重叠一个新轴,我也试图从现有问题(例如 this one)中获得灵感:这两种方法都没有用,因为不幸的是,我对 python 的了解接近于零。
有没有一种简单的方法可以在不修改现有代码的情况下绘制(未变形的)椭圆?
answer in the Q&A you linked to 确实为您的问题提供了解决方案。您想要使用那里记录的复合转换。我已经修改了你脚本的相关部分来做到这一点。
注意:使用此方法所需的一个更改是椭圆的高度以轴坐标(我使用 0.5 的高度)而不是数据坐标给出(height=10
).可能有一种方法可以用另一个转换在数据坐标中给出它,但我没有在这里包括它。我还稍微移动了椭圆中心,使其以两条红线为中心。
from matplotlib.transforms import ScaledTranslation
# Ellipse centre coordinates
x, y = 3, 8
# use the axis scale tform to figure out how far to translate
ell_offset = ScaledTranslation(x, y, ax.transScale)
# construct the composite tform
ell_tform = ell_offset + ax.transLimits + ax.transAxes
# Create the ellipse centred on the origin, apply the composite tform
ax.add_patch(Ellipse(xy=(0, 0), width=0.2, height=0.5, color="grey", fill=False, lw=1, zorder=5, transform=ell_tform))
我想使用 matplotlib 在半曲线图上画一个椭圆。
考虑下图和相关代码。
import numpy as np
import scipy.io
from matplotlib import pyplot as plt
from matplotlib.patches import Ellipse
plt.figure(figsize=[3.3, 3.3])
plt.rcParams.update({'font.size': 8, 'text.usetex': True})
plt.semilogy([1, 2, 3, 4], [4, 8, 12, 16], color='r')
plt.semilogy([1, 2, 3, 4], [2, 4, 6, 8], color='r')
plt.semilogy([1, 2, 3, 4], [12, 15, 20, 27], color='b')
ax = plt.gca()
plt.annotate('blue curve', 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, 15), xytext=(2.5, 22), arrowprops=dict(width=0.1, headwidth=2, headlength=2, color='grey'))
ax.add_patch(Ellipse(xy=(3, 10), width=0.2, height=10, color="grey", fill=False, lw=1, zorder=5))
plt.grid()
plt.xlabel('x')
plt.ylabel('y')
plt.savefig('filename.pdf', format='pdf')
plt.show()
如您所见,由于 y 轴上的对数刻度,椭圆变形了。我试图在自然尺度上重叠一个新轴,我也试图从现有问题(例如 this one)中获得灵感:这两种方法都没有用,因为不幸的是,我对 python 的了解接近于零。
有没有一种简单的方法可以在不修改现有代码的情况下绘制(未变形的)椭圆?
answer in the Q&A you linked to 确实为您的问题提供了解决方案。您想要使用那里记录的复合转换。我已经修改了你脚本的相关部分来做到这一点。
注意:使用此方法所需的一个更改是椭圆的高度以轴坐标(我使用 0.5 的高度)而不是数据坐标给出(height=10
).可能有一种方法可以用另一个转换在数据坐标中给出它,但我没有在这里包括它。我还稍微移动了椭圆中心,使其以两条红线为中心。
from matplotlib.transforms import ScaledTranslation
# Ellipse centre coordinates
x, y = 3, 8
# use the axis scale tform to figure out how far to translate
ell_offset = ScaledTranslation(x, y, ax.transScale)
# construct the composite tform
ell_tform = ell_offset + ax.transLimits + ax.transAxes
# Create the ellipse centred on the origin, apply the composite tform
ax.add_patch(Ellipse(xy=(0, 0), width=0.2, height=0.5, color="grey", fill=False, lw=1, zorder=5, transform=ell_tform))