-
Notifications
You must be signed in to change notification settings - Fork 64
Expand file tree
/
Copy pathscatter_animation.py
More file actions
58 lines (42 loc) · 1.42 KB
/
scatter_animation.py
File metadata and controls
58 lines (42 loc) · 1.42 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
50
51
52
53
54
55
56
57
58
"""
Scatter Animation Data
======================
Example showing animation with a scatter plot.
"""
# test_example = false
# sphinx_gallery_pygfx_docs = 'animate'
import fastplotlib as fpl
import numpy as np
# create a random distribution of 10,000 xyz coordinates
n_points = 10_000
# dimensions always have to be [n_points, xyz]
dims = (n_points, 3)
clouds_offset = 15
# create some random clouds
normal = np.random.normal(size=dims, scale=5)
# stack the data into a single array
cloud = np.vstack(
[
normal - clouds_offset,
normal,
normal + clouds_offset,
]
)
# color each of them separately
colors = ["yellow"] * n_points + ["cyan"] * n_points + ["magenta"] * n_points
# create plot
figure = fpl.Figure(size=(700, 560))
subplot_scatter = figure[0, 0]
# use an alpha value since this will be a lot of points
scatter_graphic = subplot_scatter.add_scatter(data=cloud, sizes=3, colors=colors, alpha=0.6)
def update_points(subplot):
# move every point by a small amount
deltas = np.random.normal(size=scatter_graphic.data.value.shape, loc=0, scale=0.15)
scatter_graphic.data = scatter_graphic.data.value + deltas
subplot_scatter.add_animations(update_points)
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()