Оновлення: Користувач cphyc люб’язно створив сховище Github для коду в цій відповіді (див. Тут ) і об’єднав код у пакет, який може бути встановлений за допомогою pip install matplotlib-label-lines.
Гарна картина:
У matplotlibце досить легко етикетки ізополей (або автоматично , або вручну розміщення ярликів з мишею). Здається, (поки) немає жодної еквівалентної можливості позначати серії даних таким чином! Можливо, є якась семантична причина не включення цієї функції, яка мені бракує.
Незважаючи на це, я написав наступний модуль, який приймає будь-які можливості для напівавтоматичного позначення ділянки. Для цього потрібні лише numpyпара функцій зі стандартної mathбібліотеки.
Опис
За замовчуванням поведінка labelLinesфункції полягає у рівномірному xрозміщенні міток вздовж осі ( yзвичайно, автоматично встановлюючи правильне значення). Якщо ви хочете, ви можете просто передати масив x координат кожної з міток. Ви навіть можете налаштувати розташування однієї мітки (як показано на нижньому правому графіку), а решту рівномірно розставити, якщо хочете.
Крім того, label_linesфункція не враховує рядки, яким не було призначено мітку в plotкоманді (або точніше, якщо мітка містить '_line').
Аргументи ключового слова, що передаються labelLinesабо labelLineпередаються у textвиклик функції (деякі аргументи ключових слів встановлюються, якщо код виклику вирішив не вказати).
Проблеми
- Поля, що обмежують анотації, іноді небажано втручаються в інші криві. Як показано анотаціями
1та та 10у верхньому лівому графіку. Я навіть не впевнений, що цього можна уникнути.
- Було б непогано
yіноді вказати посаду.
- Це все ще ітераційний процес отримання анотацій у потрібному місці
- Це працює лише тоді, коли
xзначення -x осі floats
Маю
- За замовчуванням
labelLinesфункція передбачає, що всі ряди даних охоплюють діапазон, визначений межами осі. Погляньте на синю криву у верхній лівій частині гарненького малюнка. Якби для xдіапазону були доступні лише дані 0.5- 1тоді ми не могли розмістити ярлик у потрібному місці (що трохи менше, ніж 0.2). Дивіться це питання на особливо неприємний приклад. Зараз код не інтелектуально визначає цей сценарій і не переставляє мітки, однак є обґрунтоване рішення. Функція labelLines приймає xvalsаргумент; список x-значень, заданих користувачем, замість лінійного розподілу за замовчуванням по ширині. Тож користувач може вирішити, який самеx-значення, що використовуються для розміщення етикеток кожного ряду даних.
Крім того, я вважаю, що це перша відповідь на виконання бонусної мети - вирівнювання міток з кривою, на якій вони перебувають. :)
label_lines.py:
from math import atan2,degrees
import numpy as np
#Label line with line2D label data
def labelLine(line,x,label=None,align=True,**kwargs):
ax = line.axes
xdata = line.get_xdata()
ydata = line.get_ydata()
if (x < xdata[0]) or (x > xdata[-1]):
print('x label location is outside data range!')
return
#Find corresponding y co-ordinate and angle of the line
ip = 1
for i in range(len(xdata)):
if x < xdata[i]:
ip = i
break
y = ydata[ip-1] + (ydata[ip]-ydata[ip-1])*(x-xdata[ip-1])/(xdata[ip]-xdata[ip-1])
if not label:
label = line.get_label()
if align:
#Compute the slope
dx = xdata[ip] - xdata[ip-1]
dy = ydata[ip] - ydata[ip-1]
ang = degrees(atan2(dy,dx))
#Transform to screen co-ordinates
pt = np.array([x,y]).reshape((1,2))
trans_angle = ax.transData.transform_angles(np.array((ang,)),pt)[0]
else:
trans_angle = 0
#Set a bunch of keyword arguments
if 'color' not in kwargs:
kwargs['color'] = line.get_color()
if ('horizontalalignment' not in kwargs) and ('ha' not in kwargs):
kwargs['ha'] = 'center'
if ('verticalalignment' not in kwargs) and ('va' not in kwargs):
kwargs['va'] = 'center'
if 'backgroundcolor' not in kwargs:
kwargs['backgroundcolor'] = ax.get_facecolor()
if 'clip_on' not in kwargs:
kwargs['clip_on'] = True
if 'zorder' not in kwargs:
kwargs['zorder'] = 2.5
ax.text(x,y,label,rotation=trans_angle,**kwargs)
def labelLines(lines,align=True,xvals=None,**kwargs):
ax = lines[0].axes
labLines = []
labels = []
#Take only the lines which have labels other than the default ones
for line in lines:
label = line.get_label()
if "_line" not in label:
labLines.append(line)
labels.append(label)
if xvals is None:
xmin,xmax = ax.get_xlim()
xvals = np.linspace(xmin,xmax,len(labLines)+2)[1:-1]
for line,x,label in zip(labLines,xvals,labels):
labelLine(line,x,label,align,**kwargs)
Тестовий код для створення гарної картинки вище:
from matplotlib import pyplot as plt
from scipy.stats import loglaplace,chi2
from labellines import *
X = np.linspace(0,1,500)
A = [1,2,5,10,20]
funcs = [np.arctan,np.sin,loglaplace(4).pdf,chi2(5).pdf]
plt.subplot(221)
for a in A:
plt.plot(X,np.arctan(a*X),label=str(a))
labelLines(plt.gca().get_lines(),zorder=2.5)
plt.subplot(222)
for a in A:
plt.plot(X,np.sin(a*X),label=str(a))
labelLines(plt.gca().get_lines(),align=False,fontsize=14)
plt.subplot(223)
for a in A:
plt.plot(X,loglaplace(4).pdf(a*X),label=str(a))
xvals = [0.8,0.55,0.22,0.104,0.045]
labelLines(plt.gca().get_lines(),align=False,xvals=xvals,color='k')
plt.subplot(224)
for a in A:
plt.plot(X,chi2(5).pdf(a*X),label=str(a))
lines = plt.gca().get_lines()
l1=lines[-1]
labelLine(l1,0.6,label=r'$Re=${}'.format(l1.get_label()),ha='left',va='bottom',align = False)
labelLines(lines[:-1],align=False)
plt.show()
plt.plot(x2, 3*x2**2, label="3x*x"); plt.plot(x2, 2*x2**2, label="2x*x"); plt.plot(x2, 0.5*x2**2, label="0.5x*x"); plt.plot(x2, -1*x2**2, label="-x*x"); plt.plot(x2, -2.5*x2**2, label="-2.5*x*x"); my_legend();це ставить одну з міток у верхньому лівому куті. Будь-які ідеї щодо того, як це виправити? Здається, проблема полягає в тому, що рядки розташовані занадто близько один до одного.