5

我试图在 matplotlib 中有一个旋转的文本。不幸的是，旋转似乎在显示坐标系中，而不是在数据坐标系中。那是：

import numpy as np
import matplotlib.pyplot as plt

fig = plt.figure()
ax = fig.add_axes([0.15, 0.1, 0.8, 0.8])
t = np.arange(0.0, 1.0, 0.01)
line, = ax.plot(t, t, color='blue', lw=2)
ax.text (0.51,0.51,"test label", rotation=45)
plt.show()

将给出一条在数据坐标系中为 45 度的线，但伴随的文本将在显示坐标系中为 45 度。即使在调整图形大小时，我也希望文本和数据对齐。我在这里看到我可以转换旋转，但这只有在不调整情节大小的情况下才有效。我试过写ax.text (0.51,0.51,"test label", transform=ax.transData, rotation=45)，但无论如何它似乎是默认值，并且对旋转没有帮助

有没有办法在数据坐标系中进行旋转？

编辑：

我有兴趣在绘制图形后调整图形大小 - 这是因为我通常会画一些东西，然后在保存图形之前先玩它

4

2 回答 2

8

您可以使用以下类来创建沿线的文本。而不是一个角度，它需要两个点 (p和pa) 作为输入。这两个点之间的连接定义了数据坐标中的角度。如果pa未给出，则使用p和xy（文本坐标）之间的连接线。
然后角度会自动更新，以使文本始终沿线定向。这甚至适用于对数刻度。

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.text as mtext
import matplotlib.transforms as mtransforms


class RotationAwareAnnotation(mtext.Annotation):
    def __init__(self, s, xy, p, pa=None, ax=None, **kwargs):
        self.ax = ax or plt.gca()
        self.p = p
        if not pa:
            self.pa = xy
        self.calc_angle_data()
        kwargs.update(rotation_mode=kwargs.get("rotation_mode", "anchor"))
        mtext.Annotation.__init__(self, s, xy, **kwargs)
        self.set_transform(mtransforms.IdentityTransform())
        if 'clip_on' in kwargs:
            self.set_clip_path(self.ax.patch)
        self.ax._add_text(self)

    def calc_angle_data(self):
        ang = np.arctan2(self.p[1]-self.pa[1], self.p[0]-self.pa[0])
        self.angle_data = np.rad2deg(ang)

    def _get_rotation(self):
        return self.ax.transData.transform_angles(np.array((self.angle_data,)), 
                            np.array([self.pa[0], self.pa[1]]).reshape((1, 2)))[0]

    def _set_rotation(self, rotation):
        pass

    _rotation = property(_get_rotation, _set_rotation)

示例用法：

fig, ax = plt.subplots()
t = np.arange(0.0, 1.0, 0.01)
line, = ax.plot(t, t, color='blue', lw=2)

ra = RotationAwareAnnotation("test label", xy=(.5,.5), p=(.6,.6), ax=ax,
                             xytext=(2,-1), textcoords="offset points", va="top")

plt.show()

在此处输入图像描述

边缘情况的替代方案

在某些情况下，沿垂直线的文本或具有高度不同的 x 和 y 单位的比例尺（例如此处），上述方法可能会失败。在这种情况下，以下将更适合。它计算屏幕坐标中的角度，而不是依赖于角度变换。

class RotationAwareAnnotation2(mtext.Annotation):
    def __init__(self, s, xy, p, pa=None, ax=None, **kwargs):
        self.ax = ax or plt.gca()
        self.p = p
        if not pa:
            self.pa = xy
        kwargs.update(rotation_mode=kwargs.get("rotation_mode", "anchor"))
        mtext.Annotation.__init__(self, s, xy, **kwargs)
        self.set_transform(mtransforms.IdentityTransform())
        if 'clip_on' in kwargs:
            self.set_clip_path(self.ax.patch)
        self.ax._add_text(self)

    def calc_angle(self):
        p = self.ax.transData.transform_point(self.p)
        pa = self.ax.transData.transform_point(self.pa)
        ang = np.arctan2(p[1]-pa[1], p[0]-pa[0])
        return np.rad2deg(ang)

    def _get_rotation(self):
        return self.calc_angle()

    def _set_rotation(self, rotation):
        pass

    _rotation = property(_get_rotation, _set_rotation)

在通常情况下，两者都会产生相同的输出。我不确定二等舱是否有任何缺点，所以我将两者都留在这里，选择你认为更合适的那个。

于 2018-11-02T02:07:13.553 回答
2

好的，从类似于您的示例的代码开始：

%pylab inline
import numpy as np
fig = plt.figure()
ax = fig.add_axes([0.15, 0.1, 0.8, 0.8])
t = np.arange(0.0, 1.0, 0.01)
line, = ax.plot(t, t, color='blue', lw=2)
ax.text(0.51,0.51,"test label", rotation=45)
plt.show()

在此处输入图像描述

正如您所指出的，文本标签未正确旋转以与线平行。

正如您所指出的，已在此链接中解释了文本对象相对于线条旋转的坐标系中的分离。解决方案是将文本旋转角度从绘图转换为屏幕坐标系，让我们看看调整绘图大小是否会导致您建议的问题：

for fig_size in [(3.0,3.0),(9.0,3.0),(3.0,9.0)]: #use different sizes, in inches
    fig2 = plt.figure(figsize=fig_size)
    ax = fig2.add_axes([0.15, 0.1, 0.8, 0.8])
    text_plot_location = np.array([0.51,0.51]) #I'm using the same location for plotting text as you did above
    trans_angle = gca().transData.transform_angles(np.array((45,)),text_plot_location.reshape((1,2)))[0]
    line, = ax.plot(t, t, color='blue', lw=2)
    ax.text(0.51,0.51,"test label", rotation=trans_angle)
    plt.show()

在此处输入图像描述 在此处输入图像描述 在此处输入图像描述

对我来说看起来不错，即使调整大小。现在，如果您使线更长并且轴限制更长，那么您当然必须调整文本绘制以出现在绘图的新中心。

于 2013-11-24T00:30:28.773 回答