| 1234567891011121314151617181920212223242526272829303132333435363738394041 |
- """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()
|
