Merge pull request #20499 from timhoffm/plot-types-pie

QuLogic · web-flow · commit b7ce415c15eb · 2021-06-23T20:09:45.000-04:00
Simplify plot types pie()
diff --git a/plot_types/basic/pie.py b/plot_types/basic/pie.py
@@ -5,7 +5,6 @@
 
 See `~matplotlib.axes.Axes.pie`.
 """
-import matplotlib as mpl
 import matplotlib.pyplot as plt
 import numpy as np
 
@@ -14,16 +13,12 @@
 
 # make data
 X = [1, 2, 3, 4]
-colors = np.zeros((len(X), 4))
-colors[:] = mpl.colors.to_rgba("C0")
-colors[:, 3] = np.linspace(0.25, 0.75, len(X))
+colors = plt.get_cmap('Blues')(np.linspace(0.2, 0.7, len(X)))
 
 # plot
 fig, ax = plt.subplots()
-ax.pie(X, colors=["white"]*len(X), radius=3, center=(4, 4),
-        wedgeprops={"linewidth": 1, "edgecolor": "white"}, frame=True)
 ax.pie(X, colors=colors, radius=3, center=(4, 4),
-        wedgeprops={"linewidth": 1, "edgecolor": "white"}, frame=True)
+       wedgeprops={"linewidth": 1, "edgecolor": "white"}, frame=True)
 
 ax.set(xlim=(0, 8), xticks=np.arange(1, 8),
        ylim=(0, 8), yticks=np.arange(1, 8))
