TL; DR -> Як можна створити легенду для лінійного графіка в
Matplotlib
'sPyPlot
без створення зайвих змінних?
Будь ласка, врахуйте графічний сценарій нижче:
if __name__ == '__main__':
PyPlot.plot(total_lengths, sort_times_bubble, 'b-',
total_lengths, sort_times_ins, 'r-',
total_lengths, sort_times_merge_r, 'g+',
total_lengths, sort_times_merge_i, 'p-', )
PyPlot.title("Combined Statistics")
PyPlot.xlabel("Length of list (number)")
PyPlot.ylabel("Time taken (seconds)")
PyPlot.show()
Як бачите, це дуже основне використання matplotlib
's PyPlot
. Це в ідеалі генерує графік, як показано нижче:
Нічого особливого, я знаю. Однак незрозуміло, які дані наводяться де (я намагаюся побудувати дані деяких алгоритмів сортування, тривалості від часу, і я хотів би переконатися, що люди знають, яка лінія). Таким чином, мені потрібна легенда, дивлячись на наступний приклад нижче ( з офіційного сайту ):
ax = subplot(1,1,1)
p1, = ax.plot([1,2,3], label="line 1")
p2, = ax.plot([3,2,1], label="line 2")
p3, = ax.plot([2,3,1], label="line 3")
handles, labels = ax.get_legend_handles_labels()
# reverse the order
ax.legend(handles[::-1], labels[::-1])
# or sort them by labels
import operator
hl = sorted(zip(handles, labels),
key=operator.itemgetter(1))
handles2, labels2 = zip(*hl)
ax.legend(handles2, labels2)
Ви побачите, що мені потрібно створити додаткову змінну ax
. Як я можу додати в свій графік легенду, не створюючи цю додаткову змінну і не зберігши простоту мого поточного сценарію?
matplotlib
, я хотів зробити все максимально простим. Крім того, якщо ви подивитесь на відповідь Роба, це набагато простіше, ніж приклад, показаний на веб-сайті. Я сподіваюся, що це допомагає.
import matplotlib.pyplot as plt
PyPlot