Python – Легенда перекрывается круговой диаграммой

Короткий ответ: вы можете использовать аргументы bbox_to_anchor loc , bbox_to_anchor и дополнительно bbox_transform и mode , чтобы расположить легенду по осям или фигуре.

---

Длительная версия:

Шаг 1: Убедитесь, что требуется легенда.

Во многих случаях никакой легенды не требуется вообще, и информация может быть выведена по контексту или цвету напрямую:

Если даже сюжет не может жить без легенды, переходите к шагу 2.

Шаг 2: Убедитесь, что круговая диаграмма необходима.

Во многих случаях круговые диаграммы не являются лучшим способом передачи информации.

Если необходимость в круговой диаграмме однозначно определена, давайте приступим к размещению легенды.

Размещение легенды

plt.legend() имеет два основных аргумента для определения положения легенды. Самым важным и само по себе достаточным является аргумент loc .
Например, plt.legend(loc="upper left") разместил легенду таким образом, чтобы она находилась в верхнем левом углу рамки. Если дальнейший аргумент не указан, это ограничивающее поле будет целыми осями.

Однако мы можем указать нашу собственную ограничительную рамку, используя аргумент bbox_to_anchor . Если bbox_to_anchor получает 2-х кортеж, например bbox_to_anchor=(1,1) это означает, что ограничивающий прямоугольник расположен в правом верхнем углу осей и не имеет степени. Затем он действует как точка, относительно которой легенда будет помещена в соответствии с аргументом loc . Затем он будет расширяться из ограничивающего прямоугольника нулевого размера. Например, если loc "upper left" , верхний левый угол легенды находится в позиции (1,1), и легенда будет расширяться вправо и вниз.

Эта концепция используется для вышеупомянутого сюжета, который говорит нам потрясающую правду о предвзятости на выборах Мисс Вселенная.

 import matplotlib.pyplot as plt import matplotlib.patches total = [100] labels = ["Earth", "Mercury", "Venus", "Mars", "Jupiter", "Saturn", "Uranus", "Neptune", "Pluto *"] plt.title('Origin of Miss Universe since 1952') plt.gca().axis("equal") pie = plt.pie(total, startangle=90, colors=[plt.cm.Set3(0)], wedgeprops = { 'linewidth': 2, "edgecolor" :"k" }) handles = [] for i, l in enumerate(labels): handles.append(matplotlib.patches.Patch(color=plt.cm.Set3((i)/8.), label=l)) plt.legend(handles,labels, bbox_to_anchor=(0.85,1.025), loc="upper left") plt.gcf().text(0.93,0.04,"* out of competition since 2006", ha="right") plt.subplots_adjust(left=0.1, bottom=0.1, right=0.75) 

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

Существует также возможность использовать 4-кортеж для bbox_to_anchor . Как использовать или интерпретировать это подробно в этом вопросе: что означает аргумент 4-элементного кортежа для 'bbox_to_anchor' в matplotlib?
и затем можно использовать аргумент mode="expand" чтобы легенда вписывалась в указанную ограничительную рамку.

Существуют некоторые полезные альтернативы этому подходу:

Использование координат фигуры

Вместо указания положения легенды в координатах осей можно использовать координаты фигур. Преимущество заключается в том, что это позволит просто разместить легенду в одном углу рисунка, не изменяя большую часть остальной части. С этой целью можно использовать аргумент bbox_transform и предоставить ему преобразование фигуры. Координаты, присвоенные bbox_to_anchor , затем интерпретируются как координаты фигуры.

 plt.legend(pie[0],labels, bbox_to_anchor=(1,0), loc="lower right", bbox_transform=plt.gcf().transFigure) 

Здесь (1,0) находится нижний правый угол фигуры. Из-за стандартных расстояний между осями и краем фигуры достаточно разместить легенду, чтобы она не перекрывалась с пирогом.

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

 title = plt.title('What slows down my computer') title.set_ha("left") plt.gca().axis("equal") pie = plt.pie(total, startangle=0) labels=["Trojans", "Viruses", "Too many open tabs", "The anti-virus software"] plt.legend(pie[0],labels, bbox_to_anchor=(1,0.5), loc="center right", fontsize=10, bbox_transform=plt.gcf().transFigure) plt.subplots_adjust(left=0.0, bottom=0.1, right=0.45) 

Сохранение файла с помощью bbox_inches="tight"

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

но затем сохраните его, используя bbox_inches="tight" для savefig ,

 plt.savefig("output.png", bbox_inches="tight") 

Это создаст большую фигуру, которая сидит вокруг содержимого холста:

Здесь представлен сложный подход, который позволяет помещать легенду плотно внутри фигуры, без изменения размера фигуры: Создание фигуры с точным размером и без дополнений (и легенда вне осей)

Использование подзаголовков

Альтернативой является использование подзаговоров для резервирования места для легенды. В этом случае одна подзапись может принимать круговую диаграмму, другой подзаговор содержит легенду. Это показано ниже.

 fig = plt.figure(4, figsize=(3,3)) ax = fig.add_subplot(211) total = [4,3,2,81] labels = ["tough working conditions", "high risk of accident", "harsh weather", "it's not allowed to watch DVDs"] ax.set_title('What people know about oil rigs') ax.axis("equal") pie = ax.pie(total, startangle=0) ax2 = fig.add_subplot(212) ax2.axis("off") ax2.legend(pie[0],labels, loc="center")