在matplotlib

时间:2017-05-12 20:56:00

标签: python matplotlib rotation geometry

我正在尝试旋转形状,然后使用以下代码在同一函数中进行翻译,但只有翻译才有效。有谁知道怎么做才能进行这两种转换?谢谢!

这是我的代码:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.patches as patches
import matplotlib as mpl
from math import *

#some helper values
p=4
theta=pi/6
x1 = p*cos(theta/2)
y1 = p*sin(theta/2)
vertices =[(-x1-p/2,0), (-p/2, y1), (p/2, y1), (x1+p/2, 0), (p/2, -y1), (-p/2, -y1)] 
midPoint = [3,4]

#set up the plot
fig = plt.figure()
ax = fig.add_subplot(111)

#function to rotate and translate the standard shape to a new position
def plot_polygon(vertices, midPoint, theta):
    polygon = patches.Polygon(vertices, color="red", alpha=0.50) 
    r = mpl.transforms.Affine2D().rotate(theta) + ax.transData
    t = mpl.transforms.Affine2D().translate(midPoint[0],midPoint[1]) + ax.transData
    polygon.set_transform(r)
    polygon.set_transform(t)
    ax.add_patch(polygon)

plot_polygon(vertices, midPoint, theta)

plt.xlim(-30, 30)
plt.ylim(-30, 30)

plt.grid(True)

plt.show()

1 个答案:

答案 0 :(得分:1)

如果您设置a = 3及更晚a = 4,则a4而非.set_transform

通过set_transform设置变换,你覆盖先前设置的变换,无论最后出现什么变化都是使用的变换。

因此,您需要使用组合转换对r = mpl.transforms.Affine2D().rotate(theta) t = mpl.transforms.Affine2D().translate(midPoint[0],midPoint[1]) tra = r + t + ax.transData polygon.set_transform(tra) 进行一次调用,例如

{{1}}