Переміщення легенди про matplotlib за межами осі робить її обрізаною фігурною коробкою

Мені знайомі такі питання:

Matplotlib savefig з легендою поза сюжетом

Як викласти легенду з сюжету

Здається, що відповіді на ці запитання мають розкіш у тому, що ми можемо пограбувати з точним скороченням осі, щоб легенда підходила.

Однак скорочення осей не є ідеальним рішенням, оскільки це робить меншими дані, що робить його важче інтерпретувати; особливо коли його складний і багато чого відбувається ... отже, потрібна велика легенда

Приклад складної легенди в документації демонструє необхідність цього, оскільки легенда в їх сюжеті фактично повністю затьмарює кілька точок даних.

http://matplotlib.sourceforge.net/users/legend_guide.html#legend-of-complex-plots

Що я хотів би зробити, це динамічно розширити розмір фігурного поля, щоб вмістити розширюється легенду фігури.

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(-2*np.pi, 2*np.pi, 0.1)
fig = plt.figure(1)
ax = fig.add_subplot(111)
ax.plot(x, np.sin(x), label='Sine')
ax.plot(x, np.cos(x), label='Cosine')
ax.plot(x, np.arctan(x), label='Inverse tan')
lgd = ax.legend(loc=9, bbox_to_anchor=(0.5,0))
ax.grid('on')

Зверніть увагу, як остаточний ярлик "Зворотний загар" насправді знаходиться поза рамкою фігури (і виглядає погано відсіченим - не якість публікації!)

Нарешті, мені сказали, що це нормальна поведінка в R та LaTeX, тому я трохи розгублений, чому це так складно в python ... Чи є історична причина? Чи настільки бідний Матлаб з цього питання?

У мене (лише трохи) довша версія цього коду на пастібі http://pastebin.com/grVjc007

Вибачте EMS, але я фактично отримав ще одну відповідь зі списку розсилки matplotlib (спасибі виходить Бенджаміну Корене).

Я шукаю код коригування виклику savefig для:

fig.savefig('samplefigure', bbox_extra_artists=(lgd,), bbox_inches='tight')
#Note that the bbox_extra_artists must be an iterable

Це, мабуть, схоже на виклик у тісному_лаут, але замість цього ви дозволяєте savefig враховувати додаткових виконавців у розрахунку. Це фактично змінило розмір вікна фігури за бажанням.

import matplotlib.pyplot as plt
import numpy as np

plt.gcf().clear()
x = np.arange(-2*np.pi, 2*np.pi, 0.1)
fig = plt.figure(1)
ax = fig.add_subplot(111)
ax.plot(x, np.sin(x), label='Sine')
ax.plot(x, np.cos(x), label='Cosine')
ax.plot(x, np.arctan(x), label='Inverse tan')
handles, labels = ax.get_legend_handles_labels()
lgd = ax.legend(handles, labels, loc='upper center', bbox_to_anchor=(0.5,-0.1))
text = ax.text(-0.2,1.05, "Aribitrary text", transform=ax.transAxes)
ax.set_title("Trigonometry")
ax.grid('on')
fig.savefig('samplefigure', bbox_extra_artists=(lgd,text), bbox_inches='tight')

Це дає:

[ред.] Метою цього питання було повністю уникнути використання довільних розміщень координат довільного тексту, як це було традиційним рішенням цих проблем. Незважаючи на це, численні редагування останнім часом наполягають на їх внесенні, часто таким чином, що призвело до появи помилки коду. Зараз я виправив проблеми і прилаштував довільний текст, щоб показати, як вони також розглядаються в алгоритмі bbox_extra_artists.

Додано: Я знайшов щось, що повинно зробити трюк одразу, але решта коду нижче також пропонує альтернативу.

Використовуйте subplots_adjust()функцію для переміщення нижньої частини субплоту вгору:

fig.subplots_adjust(bottom=0.2) # <-- Change the 0.02 to work for your plot.

Тоді пограйте зі зміщенням в bbox_to_anchorчастині легенди команди команди, щоб дістати поле легенди там, де ви цього хочете. Деякі поєднання налаштування figsizeта використання параметра subplots_adjust(bottom=...)повинні створити якісний сюжет для вас.

Альтернатива: я просто змінив лінію:

fig = plt.figure(1)

до:

fig = plt.figure(num=1, figsize=(13, 13), dpi=80, facecolor='w', edgecolor='k')

і змінився

lgd = ax.legend(loc=9, bbox_to_anchor=(0.5,0))

до

lgd = ax.legend(loc=9, bbox_to_anchor=(0.5,-0.02))

і він добре відображається на моєму екрані (24-дюймовий ЕКР-монітор).

Тут figsize=(M,N)встановлено вікно фігури M на дюйм на M дюймів. Просто пограйте з цим, поки це не здасться вам правильним. Перетворіть його у більш масштабований формат зображення та використовуйте GIMP для редагування, якщо потрібно, або просто обріжте за допомогою viewportпараметра LaTeX, включаючи графіку.

Ось ще одне, дуже ручне рішення. Ви можете визначити розмір осі, і прокладки враховуються відповідно (включаючи легенду та галочки). Сподіваюся, це комусь корисно.

Приклад (розмір осей однаковий!):

Код:

#==================================================
# Plot table

colmap = [(0,0,1) #blue
         ,(1,0,0) #red
         ,(0,1,0) #green
         ,(1,1,0) #yellow
         ,(1,0,1) #magenta
         ,(1,0.5,0.5) #pink
         ,(0.5,0.5,0.5) #gray
         ,(0.5,0,0) #brown
         ,(1,0.5,0) #orange
         ]


import matplotlib.pyplot as plt
import numpy as np

import collections
df = collections.OrderedDict()
df['labels']        = ['GWP100a\n[kgCO2eq]\n\nasedf\nasdf\nadfs','human\n[pts]','ressource\n[pts]'] 
df['all-petroleum long name'] = [3,5,2]
df['all-electric']  = [5.5, 1, 3]
df['HEV']           = [3.5, 2, 1]
df['PHEV']          = [3.5, 2, 1]

numLabels = len(df.values()[0])
numItems = len(df)-1
posX = np.arange(numLabels)+1
width = 1.0/(numItems+1)

fig = plt.figure(figsize=(2,2))
ax = fig.add_subplot(111)
for iiItem in range(1,numItems+1):
  ax.bar(posX+(iiItem-1)*width, df.values()[iiItem], width, color=colmap[iiItem-1], label=df.keys()[iiItem])
ax.set(xticks=posX+width*(0.5*numItems), xticklabels=df['labels'])

#--------------------------------------------------
# Change padding and margins, insert legend

fig.tight_layout() #tight margins
leg = ax.legend(loc='upper left', bbox_to_anchor=(1.02, 1), borderaxespad=0)
plt.draw() #to know size of legend

padLeft   = ax.get_position().x0 * fig.get_size_inches()[0]
padBottom = ax.get_position().y0 * fig.get_size_inches()[1]
padTop    = ( 1 - ax.get_position().y0 - ax.get_position().height ) * fig.get_size_inches()[1]
padRight  = ( 1 - ax.get_position().x0 - ax.get_position().width ) * fig.get_size_inches()[0]
dpi       = fig.get_dpi()
padLegend = ax.get_legend().get_frame().get_width() / dpi 

widthAx = 3 #inches
heightAx = 3 #inches
widthTot = widthAx+padLeft+padRight+padLegend
heightTot = heightAx+padTop+padBottom

# resize ipython window (optional)
posScreenX = 1366/2-10 #pixel
posScreenY = 0 #pixel
canvasPadding = 6 #pixel
canvasBottom = 40 #pixel
ipythonWindowSize = '{0}x{1}+{2}+{3}'.format(int(round(widthTot*dpi))+2*canvasPadding
                                            ,int(round(heightTot*dpi))+2*canvasPadding+canvasBottom
                                            ,posScreenX,posScreenY)
fig.canvas._tkcanvas.master.geometry(ipythonWindowSize) 
plt.draw() #to resize ipython window. Has to be done BEFORE figure resizing!

# set figure size and ax position
fig.set_size_inches(widthTot,heightTot)
ax.set_position([padLeft/widthTot, padBottom/heightTot, widthAx/widthTot, heightAx/heightTot])
plt.draw()
plt.show()
#--------------------------------------------------
#==================================================