Как добавить прямоугольник и метку на круговую диаграмму точно так, как показано на рисунке ниже

Question

Попытка создать график в точности как на рисунке ниже и хотите сохранить его в формате JPEG.У меня есть фрейм данных с именем 'custpref', как показано ниже

tov_type            count
Inpatient               7
Office Visit            6
Appointment Schedule    1
Allergy Sheet           1

Пробный код выглядит следующим образом: -

def addPieGraph():
# Create a list of colors (from iWantHue)
colors = ["#6287da","#72ac5c","#8653aa","#bb7438","#b94b75"]

# Create a pie chart
plt.pie(
    # using data total)arrests
    custpref['cnt'],
    # with the labels being officer names
    labels=custpref['tov_type'],
    # with no shadows
    shadow=False,
    # with colors
    colors=colors,
    # with the start angle at 90%
    startangle=90,
)

# View the plot drop above
plt.axis('equal')

# View the plot
plt.tight_layout()
plt.title("Top 5 Visit Types                  Total = 15 Visits")
plt.savefig(r"PieChart.png",bbox_inches="tight")
plt.show()

ожидаемый результат: - ожидаемый вид круговой диаграммы

Приведенный выше код для построения круговой диаграммы просто нуждается в помощи с легендами, стрелками, указывающими на них, счетчиком перед именами и меткой в ​​прямоугольной рамке с рамкой вокруг графика, как в ожидаемом результате.

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

Answer 1

Примечание: Хотя это не идеальное воссоздание вашей цели, я думаю, что оно достаточно близко для того, чтобы вы смогли точно настроить желаемый результат.

Я использовал gridspec для создания двух отдельных вспомогательных участков (pie_chart и title), добавил пользовательские строки аннотаций (адаптированные из документации ) и отформатировал вложенный сюжет title в черный, без видимых клещей / колючек.

Результат сюжета:

Полный использованный код:

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib import gridspec

custpref=pd.DataFrame({'tov_type':['Inpatient','Office Visit','Appointment Schedule','Allergy Sheet'],'count':[7,6,1,1]})

fig=plt.figure(figsize=(6,4))
gs1 = gridspec.GridSpec(1,1,
    left=0.1,right=0.7,
    bottom=0.1,top=0.7,
)
gs2 = gridspec.GridSpec(1,1,
    left=0.05,right=0.95,
    bottom=0.9,top=1.0,
)

pie_ax=fig.add_subplot(gs1[0])
title_ax=fig.add_subplot(gs2[0])

# Create a list of colors (from iWantHue)
colors = ["#6287da","#72ac5c","#8653aa","#bb7438","#b94b75"]

# Create a pie chart
wedges, texts = pie_ax.pie(
    # using data total)arrests
    custpref['count'],
    # with no shadows
    shadow=False,
    # with colors
    colors=colors,
    # with the start angle at 90%
    startangle=90,
)

bbox_props = dict(boxstyle="square,pad=0.3", fc="w", ec="k", lw=0.72)
kw = dict(xycoords='data', textcoords='data', arrowprops=dict(arrowstyle="-"), zorder=0, va="center")

for i, p in enumerate(wedges):
    ang = (p.theta2 - p.theta1)/2. + p.theta1
    y = np.sin(np.deg2rad(ang))
    x = np.cos(np.deg2rad(ang))
    horizontalalignment = {-1: "right", 1: "left"}[int(np.sign(x))]
    connectionstyle = "angle,angleA=0,angleB={}".format(ang)
    kw["arrowprops"].update({"connectionstyle": connectionstyle,"color":colors[i]})
    pie_ax.annotate(custpref['tov_type'][i], xy=(x, y), xytext=(1.35*np.sign(x), 1.4*y),
                 horizontalalignment=horizontalalignment, **kw)

# View the plot drop above
pie_ax.axis('equal')


title_ax.set_facecolor('k')

title_ax.text(0.5,0.5,"Top 5 Visit Types                  Total = 15 Visits",
    ha="center",va="center",transform=title_ax.transAxes,color="w")

for side in ['top', 'bottom', 'left', 'right']:
    title_ax.spines[side].set_visible(False)
title_ax.axes.get_xaxis().set_visible(False)    
title_ax.axes.get_yaxis().set_visible(False)    


plt.savefig(r"PieChart.png",bbox_inches="tight")
plt.show()