Observação

Clique aqui para baixar o código de exemplo completo

Rotular uma torta e um donut #

Bem-vindo à padaria Matplotlib. Criaremos um gráfico de pizza e de rosquinha por meio de e mostraremos como rotulá-los com a e também com .pie methodlegendannotations

Como de costume, começaríamos definindo as importações e criaríamos uma figura com subtramas. Agora é a vez da torta. Começando com uma receita de torta, criamos os dados e uma lista de rótulos a partir dela.

Podemos fornecer uma função para o autopctargumento, que expandirá a rotulagem automática de porcentagem mostrando valores absolutos; calculamos o último a partir de dados relativos e a soma conhecida de todos os valores.

Em seguida, criamos a torta e armazenamos os objetos retornados para mais tarde. O primeiro elemento retornado da tupla retornada é uma lista de fatias. Esses são matplotlib.patches.Wedgepatches, que podem ser usados ​​diretamente como alças para uma legenda. Podemos usar o bbox_to_anchorargumento da legenda para posicionar a legenda fora da torta. Aqui usamos as coordenadas dos eixos juntamente com a localização ; ou seja, o ponto central esquerdo da legenda estará no ponto central esquerdo da caixa delimitadora, estendendo-se de a nas coordenadas dos eixos.(1, 0, 0.5, 1)"center left"(1, 0)(1.5, 1)

import numpy as np
import matplotlib.pyplot as plt

fig, ax = plt.subplots(figsize=(6, 3), subplot_kw=dict(aspect="equal"))

recipe = ["375 g flour",
          "75 g sugar",
          "250 g butter",
          "300 g berries"]

data = [float(x.split()[0]) for x in recipe]
ingredients = [x.split()[-1] for x in recipe]


def func(pct, allvals):
    absolute = int(np.round(pct/100.*np.sum(allvals)))
    return "{:.1f}%\n({:d} g)".format(pct, absolute)


wedges, texts, autotexts = ax.pie(data, autopct=lambda pct: func(pct, data),
                                  textprops=dict(color="w"))

ax.legend(wedges, ingredients,
          title="Ingredients",
          loc="center left",
          bbox_to_anchor=(1, 0, 0.5, 1))

plt.setp(autotexts, size=8, weight="bold")

ax.set_title("Matplotlib bakery: A pie")

plt.show()
Padaria Matplotlib: Uma torta

Agora é a vez do donut. Começando com uma receita de donuts, transcrevemos os dados para números (convertendo 1 ovo para 50 g) e plotamos diretamente a torta. A torta? Espera... vai ser rosquinha, não é? Bem, como vemos aqui, o donut é uma torta, tendo um certo widthconjunto nas fatias, que é diferente do seu raio. É tão fácil quanto parece. Isso é feito por meio do wedgepropsargumento.

Em seguida, queremos rotular as fatias por meio de annotations. Primeiro criamos alguns dicionários de propriedades comuns, que depois podemos passar como argumento de palavra-chave. Em seguida, iteramos sobre todas as fatias e para cada

  • calcule o ângulo do centro da cunha,

  • a partir disso, obtenha as coordenadas do ponto naquele ângulo da circunferência,

  • determinar o alinhamento horizontal do texto, dependendo de qual lado do círculo está o ponto,

  • atualize o estilo de conexão com o ângulo obtido para que a seta de anotação aponte para fora do donut,

  • por fim, crie a anotação com todos os parâmetros previamente determinados.

fig, ax = plt.subplots(figsize=(6, 3), subplot_kw=dict(aspect="equal"))

recipe = ["225 g flour",
          "90 g sugar",
          "1 egg",
          "60 g butter",
          "100 ml milk",
          "1/2 package of yeast"]

data = [225, 90, 50, 60, 100, 5]

wedges, texts = ax.pie(data, wedgeprops=dict(width=0.5), startangle=-40)

bbox_props = dict(boxstyle="square,pad=0.3", fc="w", ec="k", lw=0.72)
kw = dict(arrowprops=dict(arrowstyle="-"),
          bbox=bbox_props, 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})
    ax.annotate(recipe[i], xy=(x, y), xytext=(1.35*np.sign(x), 1.4*y),
                horizontalalignment=horizontalalignment, **kw)

ax.set_title("Matplotlib bakery: A donut")

plt.show()
Padaria Matplotlib: Um donut

E aqui está, o donut. Observe, no entanto, que se usássemos esta receita, os ingredientes seriam suficientes para cerca de 6 donuts - produzir um donut enorme não foi testado e pode resultar em erros de cozinha.

Referências

O uso das seguintes funções, métodos, classes e módulos é mostrado neste exemplo:

Galeria gerada por Sphinx-Gallery