Построение круговой диаграммы в matplotlib под определенным углом с трещинами на клиньях - PullRequest
Новая
       33

Построение круговой диаграммы в matplotlib под определенным углом с трещинами на клиньях

2 голосов
/ 10 февраля 2012

Я строю круговую диаграмму с помощью 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")

Это дает следующую круговую диаграмму. PieChart 1

Однако я хочу начать круговую диаграмму с первого клина сверху, единственное решение, которое я мог найти для этого, это использовать этот код

Однако при использовании этого, как показано ниже, 

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")

Это дает следующую круговую диаграмму

enter image description here

Однако это не выводит трещины на клиньях, как на предыдущей круговой диаграмме. Я пробовал несколько разных вещей, но я не в состоянии сохранить трещины. Как я могу запустить первый клин в полдень и вывести трещины на клиньях ??

1 Ответ

2 голосов
/ 10 февраля 2012

Обычно я бы не советовал менять исходный текст инструмента, но это легко исправить снаружи и легко внутри.Итак, вот что я бы сделал, если бы вам нужно, чтобы это работало прямо сейчас (тм), а иногда вы делаете ..

В файле 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):

т.е. просто добавьте start_angle=None в конец аргументов.

Затем добавьте пять строк, заключенных в квадратные скобки с помощью "# добавление". 

    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)

производит

enter image description here

Обратите внимание, что в целом вы должны избегать этого, исправляя источник, который я имею в виду, но бывают случаи, когдаПрошлое, когда я был в срок и просто хотел что-то сейчас (тм), так что вы идете ..

...