在 matplotlib 中以特定角度绘制饼图，并在楔形上绘制分形

Question

我使用以下代码使用 matplotlib 绘制饼图：

ax = axes([0.1, 0.1, 0.6, 0.6])
labels = 'Twice Daily', 'Daily', '3-4 times per week', 'Once per week','Occasionally'
fracs = [20,50,10,10,10]

explode=(0, 0, 0, 0,0.1)
patches, texts, autotexts = ax.pie(fracs, labels=labels, explode = explode,         
                             autopct='%1.1f%%', shadow =True)
proptease = fm.FontProperties()
proptease.set_size('xx-small')
setp(autotexts, fontproperties=proptease)
setp(texts, fontproperties=proptease)
rcParams['legend.fontsize'] = 7.0
savefig("pie1")

这会生成以下饼图。

但是，我想以顶部的第一个楔形开始饼图，这是我能找到的唯一解决方案这是使用 此代码

但是，按如下方式使用它，

from pylab import *
from matplotlib import font_manager as fm
from matplotlib.transforms import Affine2D
from matplotlib.patches import Circle, Wedge, Polygon
import numpy as np

fig = plt.figure()
ax = fig.add_subplot(111)

labels = 'Twice Daily', 'Daily', '3-4 times per week', 'Once per week','Occasionally'
fracs = [20,50,10,10,10]

 wedges, plt_labels = ax.pie(fracs, labels=labels)
 ax.axis('equal')

 starting_angle = 90
 rotation = Affine2D().rotate(np.radians(starting_angle))

for wedge, label in zip(wedges, plt_labels):
  label.set_position(rotation.transform(label.get_position()))
  if label._x > 0:
    label.set_horizontalalignment('left')
  else:
    label.set_horizontalalignment('right')

  wedge._path = wedge._path.transformed(rotation)

plt.savefig("pie2")

会生成以下饼图

但是，这不会打印压裂在如前面的饼图所示的楔形。我尝试了一些不同的方法，但我无法保存压裂。如何在中午启动第一个楔块并在楔块上显示压裂？

Answer 1

通常我不建议更改工具的源代码，但在外部修复此问题很麻烦，而在内部修复它很容易。因此，如果您需要它立即工作（tm），有时您会这样做，那么我会这样做。

在文件 matplotlib/axes.py 中，将 pie 函数的声明更改为

def pie(self, x, explode=None, labels=None, colors=None,
        autopct=None, pctdistance=0.6, shadow=False,
        labeldistance=1.1, start_angle=None):

i.e.只需将 start_angle=None 添加到参数末尾即可。

然后添加用“#addition”括起来的五行。

    for frac, label, expl in cbook.safezip(x,labels, explode):
        x, y = center
        theta2 = theta1 + frac
        thetam = 2*math.pi*0.5*(theta1+theta2)

        # addition begins here
        if start_angle is not None and i == 0:
            dtheta = (thetam - start_angle)/(2*math.pi)
            theta1 -= dtheta
            theta2 -= dtheta
            thetam = start_angle
        # addition ends here

        x += expl*math.cos(thetam)
        y += expl*math.sin(thetam)

然后，如果 start_angle 为 None，则不会发生任何事情，但如果 start_angle 有一个值，则该位置就是第一个切片（在本例中为 20%）的中心位置。例如，

patches, texts, autotexts = ax.pie(fracs, labels=labels, explode = explode,         
                             autopct='%1.1f%%', shadow =True, start_angle=0.75*pi)

生成

请注意，通常您应该避免这样做，我的意思是修补源代码，但是有过去有时我会在最后期限前完成任务，只是现在就想要一些东西，所以就这样吧..