我们可以通过 pie() 方法创建一个饼图和制作甜甜圈的菜谱图,并展示如何用图例和注释标记它们。
首先定义导入并创建一个带有子块的图,然后开始绘制饼图,为其创建数据和标签列表。
我们可以为 autoct 参数提供一个函数,它将通过显示绝对值来扩展自动百分比标记;我们根据相对数据和所有值的已知和来计算后者。然后创建饼图并存储返回的对象以备以后使用。返回元组的第一个返回元素是楔块列表。这些是matplotlib.patches.Wedge patches,可以直接用作图例的句柄。我们可以使用legend的bboxto锚定参数将legend定位到馅饼之外。这里我们使用轴坐标(1,0,0.5,1)和位置“中心左”;即图例的左中心点将位于边界框的左中心点,在轴坐标中从(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(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()
现在绘制甜甜圈食谱,我们将数据转换成数字(将1个鸡蛋转换成50克),然后直接绘制馅饼图。正如我们在这里看到的,甜甜圈是一个饼图,根据不同的半径,把一定的宽度设置为楔形,通过wedgeprops参数完成。
然后通过注释来标记楔块,首先创建一些具有公共属性的字典,可以将其作为关键字参数传递。然后我们遍历所有的楔块:
计算楔块中心的角度,
从中得到圆周上那个角度点的坐标,
确定文本的水平对齐方式,具体取决于点位于圆的哪一侧,
使用获得的角度更新连接样式,使注释箭头从圆环向外指向,
最后,使用所有先前确定的参数创建注释。
import numpy as np
import matplotlib.pyplot as plt
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()