-
Notifications
You must be signed in to change notification settings - Fork 64
Expand file tree
/
Copy pathline_cmap.py
More file actions
49 lines (38 loc) · 1.1 KB
/
line_cmap.py
File metadata and controls
49 lines (38 loc) · 1.1 KB
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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
"""
Line Plot Colormap
==================
Example showing basic colormapping with lines
"""
# test_example = true
# sphinx_gallery_pygfx_docs = 'screenshot'
import fastplotlib as fpl
import numpy as np
figure = fpl.Figure(size=(700, 560))
xs = np.linspace(-10, 10, 100)
# sine wave
ys = np.sin(xs)
sine = np.dstack([xs, ys])[0]
# cosine wave
ys = np.cos(xs) - 5
cosine = np.dstack([xs, ys])[0]
# cmap_transform from an array, so the colors on the sine line will be based on the sine y-values
sine_graphic = figure[0, 0].add_line(
data=sine,
thickness=10,
cmap="plasma",
cmap_transform=sine[:, 1]
)
# qualitative colormaps, useful for cluster labels or other types of categorical labels
labels = [0] * 25 + [5] * 10 + [1] * 35 + [2] * 30
cosine_graphic = figure[0, 0].add_line(
data=cosine,
thickness=10,
cmap="tab10",
cmap_transform=labels
)
figure.show()
# NOTE: `if __name__ == "__main__"` is NOT how to use fastplotlib interactively
# please see our docs for using fastplotlib interactively in ipython and jupyter
if __name__ == "__main__":
print(__doc__)
fpl.loop.run()