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
|
r"""
Contour plots
=============
Contour plot support is limited to picking the individual
:class:`~matplotlib.collections.LineCollection`\s, which are directly
registered with the axes and thus picked up by `mplcursors.cursor`
(:class:`~matplotlib.contour.QuadContourSet`\s are not even artists, which make
them hard to handle without additional special-casing). It remains possible to
retrieve the ``z`` value and add it manually to the annotation, though.
"""
import numpy as np
import matplotlib.pyplot as plt
import mplcursors
np.random.seed(42)
fig, ax = plt.subplots()
cf = ax.contour(np.random.random((10, 10)))
cursor = mplcursors.cursor()
@cursor.connect("add")
def on_add(sel):
ann = sel.annotation
# `cf.collections.index(sel.artist)` is the index of the selected line
# among all those that form the contour plot.
# `cf.cvalues[...]` is the corresponding value.
ann.set_text("{}\nz={:.3g}".format(
ann.get_text(), cf.cvalues[cf.collections.index(sel.artist)]))
plt.show()
|
