有没有办法将matplotlib图旋转45度？

Question

我正在寻找一种方法将matplotlib-pyplot(Python库)中生成的绘图旋转45度(例如,如果不是方形,你会有钻石形状),任何人都知道是否可以这样做？

我能想到的一种方法是对所有数据使用旋转滤镜,使其显示为旋转,但绘图本身仍将处于原始方向.

我希望能够使用matplotlib交互功能,因此保存为图像然后旋转将无法正常工作.

此外,我想使用pyplot函数绘制绘图,因此使用不同的库进行绘图并不是一个理想的解决方案.

Answer 1

好的,目前我找到的唯一部分解决方案是对绘图应用旋转.这允许使用matplotlib/pyplot提供的交互式界面.

对于像plot()和scatter()这样的点图,这是微不足道的,但我对旋转imshow()特别感兴趣.此链接讨论了可能已用于此任务的transform关键字,但它显然无法正常工作.

幸运的是,我找到了使用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))