python - كيفية وضع أسطورة من المؤامرة

إجابة مختصرة: يمكنك استخدام bbox_to_anchor + bbox_extra_artists + bbox_inches='tight' .

إجابة أطول: يمكنك استخدام bbox_to_anchor لتحديد موقع مربع bbox_to_anchor يدويًا ، كما أشار بعض الأشخاص الآخرين في الإجابات.

ومع ذلك ، فإن المشكلة المعتادة هي أن صندوق الأسطورة يتم اقتصاصه ، على سبيل المثال:

import matplotlib.pyplot as plt

# data 
all_x = [10,20,30]
all_y = [[1,3], [1.5,2.9],[3,2]]

# Plot
fig = plt.figure(1)
ax = fig.add_subplot(111)
ax.plot(all_x, all_y)

# Add legend, title and axis labels
lgd = ax.legend( [ 'Lag ' + str(lag) for lag in all_x], loc='center right', bbox_to_anchor=(1.3, 0.5))
ax.set_title('Title')
ax.set_xlabel('x label')
ax.set_ylabel('y label')

fig.savefig('image_output.png', dpi=300, format='png')

لمنع المربع الأسطورة من الحصول على الاقتصاص ، عند حفظ الشكل ، يمكنك استخدام المعلمات bbox_extra_artists و bbox_inches لطلب savefig لتضمين العناصر التي تم اقتصاصها في الصورة المحفوظة:

fig.savefig('image_output.png', bbox_extra_artists=(lgd,), bbox_inches='tight')

مثال (قمت فقط بتغيير السطر الأخير لإضافة معلمتين إلى fig.savefig() ):

import matplotlib.pyplot as plt

# data 
all_x = [10,20,30]
all_y = [[1,3], [1.5,2.9],[3,2]]

# Plot
fig = plt.figure(1)
ax = fig.add_subplot(111)
ax.plot(all_x, all_y)

# Add legend, title and axis labels
lgd = ax.legend( [ 'Lag ' + str(lag) for lag in all_x], loc='center right', bbox_to_anchor=(1.3, 0.5))
ax.set_title('Title')
ax.set_xlabel('x label')
ax.set_ylabel('y label')    

fig.savefig('image_output.png', dpi=300, format='png', bbox_extra_artists=(lgd,), bbox_inches='tight')

أتمنى لو أن matplotlib قد يسمح أصلاً بالوصول إلى موقع خارجي لمربع الأسطورة كما يفعل Matlab :

figure
x = 0:.2:12;
plot(x,besselj(1,x),x,besselj(2,x),x,besselj(3,x));
hleg = legend('First','Second','Third',...
              'Location','NorthEastOutside')
% Make the text of the legend italic and color it brown
set(hleg,'FontAngle','italic','TextColor',[.3,.2,.1])

إليك حلاً آخر ، مشابهًا لإضافة bbox_extra_artists و bbox_inches ، حيث لا يلزم أن يكون لديك فنانين إضافيين في نطاق اتصال savefig . لقد توصلت إلى هذا منذ أن أنشأت معظم مؤامراتي الداخلية.

بدلاً من إضافة جميع الإضافات إلى المربع المحيط عندما تريد كتابتها ، يمكنك إضافتها مسبقًا إلى فناني Figure . باستخدام شيء مماثل لإجابة فرانك Dernoncourt أعلاه :

import matplotlib.pyplot as plt

# data 
all_x = [10,20,30]
all_y = [[1,3], [1.5,2.9],[3,2]]

# plotting function
def gen_plot(x, y):
    fig = plt.figure(1)
    ax = fig.add_subplot(111)
    ax.plot(all_x, all_y)
    lgd = ax.legend( [ "Lag " + str(lag) for lag in all_x], loc="center right", bbox_to_anchor=(1.3, 0.5))
    fig.artists.append(lgd) # Here's the change
    ax.set_title("Title")
    ax.set_xlabel("x label")
    ax.set_ylabel("y label")
    return fig

# plotting
fig = gen_plot(all_x, all_y)

# No need for `bbox_extra_artists`
fig.savefig("image_output.png", dpi=300, format="png", bbox_inches="tight")

وهنا المؤامرة ولدت.

إنشاء خصائص الخط

from matplotlib.font_manager import FontProperties

fontP = FontProperties()
fontP.set_size('small')
legend([plot1], "title", prop=fontP)

عمل شيء على طول هذه الخطوط بالنسبة لي. بدءًا من جزء من التعليمة البرمجية المأخوذة من جو ، تعدّل هذه الطريقة عرض النافذة ليتلاءم مع وسيلة إيضاح بشكل تلقائي على يمين الشكل.

import matplotlib.pyplot as plt
import numpy as np

plt.ion()

x = np.arange(10)

fig = plt.figure()
ax = plt.subplot(111)

for i in xrange(5):
    ax.plot(x, i * x, label='$y = %ix$'%i)

# Put a legend to the right of the current axis
leg = ax.legend(loc='center left', bbox_to_anchor=(1, 0.5))

plt.draw()

# Get the ax dimensions.
box = ax.get_position()
xlocs = (box.x0,box.x1)
ylocs = (box.y0,box.y1)

# Get the figure size in inches and the dpi.
w, h = fig.get_size_inches()
dpi = fig.get_dpi()

# Get the legend size, calculate new window width and change the figure size.
legWidth = leg.get_window_extent().width
winWidthNew = w*dpi+legWidth
fig.set_size_inches(winWidthNew/dpi,h)

# Adjust the window size to fit the figure.
mgr = plt.get_current_fig_manager()
mgr.window.wm_geometry("%ix%i"%(winWidthNew,mgr.window.winfo_height()))

# Rescale the ax to keep its original size.
factor = w*dpi/winWidthNew
x0 = xlocs[0]*factor
x1 = xlocs[1]*factor
width = box.width*factor
ax.set_position([x0,ylocs[0],x1-x0,ylocs[1]-ylocs[0]])

plt.draw()

كان الحل الذي عمل بالنسبة لي عندما كان لدي أسطورة ضخمة هو استخدام تخطيط صورة فارغ إضافي. في المثال التالي ، قمت بعمل 4 صفوف وفي الجزء السفلي ، أرسم الصورة مع إزاحة عن الأسطورة (bbox_to_anchor) في الأعلى لا تحصل على قطع.

f = plt.figure()
ax = f.add_subplot(414)
lgd = ax.legend(loc='upper left', bbox_to_anchor=(0, 4), mode="expand", borderaxespad=0.3)
ax.autoscale_view()
plt.savefig(fig_name, format='svg', dpi=1200, bbox_extra_artists=(lgd,), bbox_inches='tight')

لا أعرف ما إذا كنت قد قمت بالفعل بتصنيف مشكلتك ... ربما نعم ، ولكن ... أنا ببساطة استخدمت السلسلة "خارج" للموقع ، كما هو الحال في matlab. لقد استوردت pylab من matplotlib. انظر الكود على النحو التالي:

from matplotlib as plt
from matplotlib.font_manager import FontProperties
...
...
t = A[:,0]
sensors = A[:,index_lst]

for i in range(sensors.shape[1]):
    plt.plot(t,sensors[:,i])

plt.xlabel('s')
plt.ylabel('°C')
lgd = plt.legend(b,loc='center left', bbox_to_anchor=(1, 0.5),fancybox = True, shadow = True)

انقر لرؤية المؤامرة

لوضع وسيلة الإيضاح خارج منطقة الرسم ، استخدم loc و bbox_to_anchor الكلمات الأساسية من وسيلة الإيضاح (). على سبيل المثال ، سيضع التعليمة البرمجية التالية وسيلة الإيضاح على يمين ناحية الرسم:

legend(loc="upper left", bbox_to_anchor=(1,1))

لمزيد من المعلومات ، راجع دليل وسيلة الإيضاح

هنا مثال من البرنامج التعليمي matplotlib وجدت هنا . هذا هو أحد الأمثلة الأكثر بساطة ولكنني أضفت الشفافية إلى وسيلة الإيضاح وإضافة plt.show () حتى تتمكن من لصقها في الغلاف التفاعلي والحصول على نتيجة:

import matplotlib.pyplot as plt
p1, = plt.plot([1, 2, 3])
p2, = plt.plot([3, 2, 1])
p3, = plt.plot([2, 3, 1])
plt.legend([p2, p1, p3], ["line 1", "line 2", "line 3"]).get_frame().set_alpha(0.5)
plt.show()

هناك عدد من الطرق لفعل ما تريد. للإضافة إلى ما ذكرهinalis وNavi مسبقًا ، يمكنك استخدام الوسيطة الكلمة الأساسية bbox_to_anchor الإيضاح بشكل جزئي خارج المحاور و / أو تقليل حجم الخط.

قبل أن تفكر في تقليل حجم الخط (الذي يمكن أن يجعل الأمور صعبًا للقراءة) ، حاول اللعب مع وضع الأسطورة في أماكن مختلفة:

لذا ، لنبدأ بمثال عام:

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(10)

fig = plt.figure()
ax = plt.subplot(111)

for i in xrange(5):
    ax.plot(x, i * x, label='$y = %ix$' % i)

ax.legend()

plt.show()

إذا فعلنا نفس الشيء ، ولكن bbox_to_anchor الوسيطة الكلمة الرئيسية bbox_to_anchor يمكننا نقل الأسطورة قليلاً خارج حدود المحاور:

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(10)

fig = plt.figure()
ax = plt.subplot(111)

for i in xrange(5):
    ax.plot(x, i * x, label='$y = %ix$' % i)

ax.legend(bbox_to_anchor=(1.1, 1.05))

plt.show()

وبالمثل ، يمكنك جعل الأسطورة أكثر أفقيًا و / أو وضعها في أعلى الشكل (أنا أيضًا أقوم بتشغيل الزوايا الدائرية والظل المسقط البسيط):

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(10)

fig = plt.figure()
ax = plt.subplot(111)

for i in xrange(5):
    line, = ax.plot(x, i * x, label='$y = %ix$'%i)

ax.legend(loc='upper center', bbox_to_anchor=(0.5, 1.05),
          ncol=3, fancybox=True, shadow=True)
plt.show()

بدلاً من ذلك ، يمكنك تقليص عرض المخطط الحالي ووضع الأسطورة بالكامل خارج محور الشكل:

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(10)

fig = plt.figure()
ax = plt.subplot(111)

for i in xrange(5):
    ax.plot(x, i * x, label='$y = %ix$'%i)

# Shrink current axis by 20%
box = ax.get_position()
ax.set_position([box.x0, box.y0, box.width * 0.8, box.height])

# Put a legend to the right of the current axis
ax.legend(loc='center left', bbox_to_anchor=(1, 0.5))

plt.show()

وبطريقة مشابهة ، يمكنك تقليص المؤامرة عموديًا ، ووضع الأسطورة الأفقية في الأسفل:

import matplotlib.pyplot as plt
import numpy as np

x = np.arange(10)

fig = plt.figure()
ax = plt.subplot(111)

for i in xrange(5):
    line, = ax.plot(x, i * x, label='$y = %ix$'%i)

# Shrink current axis's height by 10% on the bottom
box = ax.get_position()
ax.set_position([box.x0, box.y0 + box.height * 0.1,
                 box.width, box.height * 0.9])

# Put a legend below current axis
ax.legend(loc='upper center', bbox_to_anchor=(0.5, -0.05),
          fancybox=True, shadow=True, ncol=5)

plt.show()

إلقاء نظرة على دليل أسطورة matplotlib . يمكنك أيضًا إلقاء نظرة على plt.figlegend() . نأمل أن يساعد قليلا ، على أي حال!

الجواب القصير : يمكنك استدعاء السحب على وسيلة الإيضاح ونقلها بشكل تبادلي أينما تريد:

ax.legend().draggable()

إجابة طويلة : إذا كنت تفضل وضع وسيلة الإيضاح بشكل تفاعلي / يدوي بدلاً من برمجياً ، يمكنك تبديل الوضع القابل للسحب من وسيلة الإيضاح بحيث يمكنك سحبها إلى أي مكان تريده. تحقق من المثال أدناه:

import matplotlib.pylab as plt
import numpy as np
#define the figure and get an axes instance
fig = plt.figure()
ax = fig.add_subplot(111)
#plot the data
x = np.arange(-5, 6)
ax.plot(x, x*x, label='y = x^2')
ax.plot(x, x*x*x, label='y = x^3')
ax.legend().draggable()
plt.show()