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
|
"""Plotting functions with error bands"""
from vedo import np, Rectangle, Text3D, Marker, Line
from vedo.pyplot import plot
# Make up same dummy data
x = np.arange(0, 6, 0.05)
y = 2+2*np.sin(2*x)/(x+1)
ye= y**2 / 10
miny = np.min(y-ye)
idx = np.argmax(y)
# Plot the two variables, return a Plot(Assembly) object:
fig = plot(
x, y,
yerrors=ye,
xtitle='time in :museconds',
ytitle='y oscillation [a.u.]',
ylim=(0.5, 5),
aspect=5/3, # plot aspect ratio (xsize/ysize)
error_band=True, # join errors on y into an error band
lc="red2", # line color
ec="red7", # error band color
padding=0, # no extra spaces around the content
grid=0, # no background grid
axes=dict(axes_linewidth=2, xyframe_line=0),
)
# Add a grey transparent rectangle to represent an exclusion region:
fig += Rectangle([1,0.5], [2.7,5], c='grey5').lighting('off')
# Add some text (set z=2 so it stays on top):
fig += Text3D("Excluded\ntime range!",
s=0.2, c='k', font="Quikhand").rotate_z(20).pos(1.3,3.6)
# Add a star marker at maximum of function (set z=0.1, so it stays on top):
fig += Marker('*', c='blue4').pos(x[idx], y[idx], 0.1)
# Add a dashed line to indicate the minimum
fig += Line((x[0], miny), (x[-1], miny)).pattern('- . -').lw(3)
fig.show(zoom='tight', mode='image', size=(900,600)).close()
|
