我正在寻找一种方法将matplotlib-pyplot(Python库)中生成的绘图旋转45度(例如,如果不是方形,你会有钻石形状),任何人都知道是否可以这样做?
我能想到的一种方法是对所有数据使用旋转滤镜,使其显示为旋转,但绘图本身仍将处于原始方向.
我希望能够使用matplotlib交互功能,因此保存为图像然后旋转将无法正常工作.
此外,我想使用pyplot函数绘制绘图,因此使用不同的库进行绘图并不是一个理想的解决方案.
解决方法:
好的,目前我找到的唯一部分解决方案是对绘图应用旋转.这允许使用matplotlib / pyplot提供的交互式界面.
对于像plot()和scatter()这样的点图,这是微不足道的,但我对旋转imshow()特别感兴趣. This链接讨论了可能已用于此任务的转换关键字,但它显然无法正常工作.
幸运的是,我找到了使用pcolormesh()的解决方法. pcolormesh()绘制四边形网格,并允许您指定角坐标.因此,答案是将相关变换应用于角坐标.但请注意,pcolormesh()的工作方式与imshow略有不同 – 它会将您的矩阵翻转.
我没有在网络上的任何地方看到这个解决方案,所以这里有一些代码为pcolormesh()/ imshow()旋转了45度:
import matplotlib.pyplot as plt
import numpy as np
def pcolormesh_45deg(C):
n = C.shape[0]
# create rotation/scaling matrix
t = np.array([[1,0.5],[-1,0.5]])
# create coordinate matrix and transform it
A = np.dot(np.array([(i[1],i[0]) for i in itertools.product(range(n,-1,-1),range(0,n+1,1))]),t)
# plot
plt.pcolormesh(A[:,1].reshape(n+1,n+1),A[:,0].reshape(n+1,n+1),np.flipud(C))
标签:python,matplotlib