1

我想用椭圆作为标记制作一个图。这是一个示例代码，它制作了一个大椭圆，现在实际上是一个圆。

#! /usr/bin/env python3.2
import numpy as np
import pylab
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages
from matplotlib.patches import Ellipse

PlotFileName="test.pdf"
pdf = PdfPages(PlotFileName)
fig=plt.figure(1)
ax1=fig.add_subplot(111)
x_lim=3
plt.xlim([0,x_lim])
plt.ylim([0,x_lim])

F=pylab.gcf()
DefSize = F.get_size_inches()

#These lines have a true angle of 45 degrees, only as a reference:
offset_along_x=x_lim-(x_lim/ax_ratio)
ax1.plot([offset_along_x/2, x_lim-(offset_along_x/2)], [0, x_lim], "b")
ax1.plot([offset_along_x/2, x_lim-(offset_along_x/2)], [x_lim, 0], "b")

e=0.0
theta=0
maj_ax=2
min_ax=maj_ax*np.sqrt(1-e**2)
xconst=(DefSize[1]/DefSize[0])*np.cos(theta*np.pi/180)-np.sin(theta*np.pi/180)
yconst=np.cos(theta*np.pi/180)+(DefSize[1]/DefSize[0])*np.sin(theta*np.pi/180)
print("xconstant= {}".format(xconst))
print("yconstant= {}".format(yconst))
ax1.add_artist(Ellipse((x_lim/2, x_lim/2), xconst*maj_ax, yconst*min_ax, angle=theta, facecolor="green", edgecolor="black",zorder=2, alpha=0.5))

pdf.savefig(fig)
pdf.close()
plt.close()

虽然在这个简化的情况下，ax1.axis("equal")会给出一个纯圆，但在我的最终情节中，这个命令会破坏整个情节（比例不相等）。所以我想制作一个通用的椭圆工具而不使用ax1.axis("equal"). 如您所见，您可以在此程序中设置主轴的偏心率和倾角。

问题： 问题似乎是我不明白 matplotlib 如何旋转它的图像。如果将theta此处的值更改为 0 或 90 以外的值，则对象将不再是圆形。输出是一个圆圈，ax1.axis("equal")现在不管值theta是多少。所以我的第一个问题是：在更改theta. 我假设一旦我解决了这个问题，它也适用于椭圆。有人可以帮我吗？我真的很感激。

4

1 回答 1

3

在 matplotlib 中，通过首先缩放到高度和宽度然后围绕其中心旋转然后平移到所需位置来设置日食位置和形式。所以旋转和缩放高度和之前可能会变坏（就像在你的脚本中一样），但很难完全正确（你可能必须缩放和反转旋转，但我的变换数学很生疏）。

如果要正确缩放椭圆的形状，则需要子类化 Ellipse 类并重新定义_recompute_transform函数：

from matplotlib import transforms

class TransformedEllipse(Ellipse):

    def __init__(self, xy, width, height, angle=0.0, fix_x = 1.0, **kwargs):
        Ellipse.__init__(self, xy, width, height, angle, **kwargs)

        self.fix_x = fix_x

    def _recompute_transform(self):

        center = (self.convert_xunits(self.center[0]),
                  self.convert_yunits(self.center[1]))
        width = self.convert_xunits(self.width)
        height = self.convert_yunits(self.height)
        self._patch_transform = transforms.Affine2D() \
            .scale(width * 0.5, height * 0.5) \
            .rotate_deg(self.angle) \
            .scale(self.fix_x, 1) \
            .translate(*center)

并像这样使用它：

fix_x = DefSize[1]/DefSize[0]/ax_ratio

ellipse = TransformedEllipse((x_lim/2.0, x_lim/2.0), maj_ax, min_ax, angle=theta, facecolor="green", edgecolor="black",zorder=2, alpha=0.5, fix_x = fix_x)

PS我认为ax_ratio是y_lim / x_lim！

于 2012-07-05T12:11:15.373 回答