круговая диаграмма: вращать этикетки по горизонтали

Question

Использование matplot для создания небольшого датчика с кодом ниже:

    group_size=[10,10,10,10,10,50]
    labels=['AAAA','BBBB','CCCC','DDDD','EEEE','']
    fig, ax = plt.subplots()
    ax.axis('equal')
    pie = ax.pie(group_size, radius=2.2, colors=['k'] ,startangle=180,counterclock=False)
    pie2 = ax.pie([10,10,10,10,10,50], radius=2,  labeldistance=0.7, labels=labels, rotatelabels = 270,
startangle=180,counterclock=False)
    plt.show()

Я пытаюсь, чтобы метки располагались горизонтально и выравнивались со средней точкой каждого клина, как это (но с текстом внутри клиньев):

Используя rotatelabel = True, я получаю следующее:

Есть идеи, как мне добиться горизонтального поворота надписей на графике?

Answer 1

Мое решение в основном похоже на принятый ответ ImportanceOfBeingErnest. Но я думаю, что описанные шаги легче понять.

import matplotlib.pyplot as plt

group_size = [10, 10, 10, 10, 10, 50]
labels = ['AAAA', 'BBBB', 'CCCC', 'DDDD', 'EEEE', '']
fig, ax = plt.subplots()
ax.axis('equal')
pie = ax.pie(group_size, radius=2.2, colors=['k'], startangle=180, counterclock=False)

# ax.pie() returns wedges, labels
# Note that rotatelabels=False; so that, at this step ...
# all labels are not rotated; they will be rotated later
wedges, labels = ax.pie([10, 10, 10, 10, 10, 50], radius=2,  \
                      labeldistance=0.85, labels=labels, rotatelabels = False, \
                      startangle=180, counterclock=False)

# do the rotation of the labels
for ea, eb in zip(wedges, labels):
    mang =(ea.theta1 + ea.theta2)/2.  # get mean_angle of the wedge
    #print(mang, eb.get_rotation())
    eb.set_rotation(mang+270)         # rotate the label by (mean_angle + 270)
    eb.set_va("center")
    eb.set_ha("center")

plt.show()

Полученное изображение:

Answer 2

Вам нужно повернуть круговые метки вручную. Для этого вы можете зациклить надписи и установить ротацию в соответствии с вашими потребностями.

group_size=[10,10,10,10,10,50]
labels=['AAAA','BBBB','CCCC','DDDD','EEEE','']
fig, ax = plt.subplots()
ax.axis('equal')
pie = ax.pie(group_size, radius=2.2, colors=['k'] ,startangle=180,counterclock=False)
pie2 = ax.pie([10,10,10,10,10,50], radius=2,  labeldistance=0.9, labels=labels, 
              rotatelabels =True, startangle=180,counterclock=False)

plt.setp(pie2[1], rotation_mode="anchor", ha="center", va="center")
for tx in pie2[1]:
    rot = tx.get_rotation()
    tx.set_rotation(rot+90+(1-rot//180)*180)

plt.show()