1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
"""Plot to test legend"""
import matplotlib.pyplot as plt
import numpy as np
import mpld3
from mpld3 import plugins
def create_plot():
fig, ax = plt.subplots()
x = np.linspace(0, 10, 100)
l1 = ax.plot(x, np.sin(x), label='sin', lw=3, alpha=0.2)
l2 = ax.plot(x, np.cos(x), label='cos', lw=3, alpha=0.2)
l3 = ax.plot(x[::5], 0.5 * np.sin(x[::5] + 2), 'ob', label='dots',
alpha=0.2)
labels = ['sin', 'cos', 'dots']
interactive_legend = plugins.InteractiveLegendPlugin([l1, l2, l3], labels)
plugins.connect(fig, interactive_legend)
ax.set_title("Interactive legend test", size=20)
return fig
def test_legend():
fig = create_plot()
html = mpld3.fig_to_html(fig)
plt.close(fig)
if __name__ == "__main__":
mpld3.show(create_plot())
|
