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
|
# -*- coding: utf-8 -*-
# Copyright (c) Vispy Development Team. All Rights Reserved.
# Distributed under the (new) BSD License. See LICENSE.txt for more info.
# vispy: gallery 2
"""
Plot different styles of ColorBar using vispy.plot
"""
from vispy import plot as vp
import numpy as np
# arg( e^(1/z) )
def exp_z_inv(x, y):
z = np.complex(x, y)
f = np.exp(1.0 / z)
return np.angle(f, deg=True)
# create a 2d grid whose elements are of exp_z_inv
def gen_image(width, height):
x_vals = np.linspace(-0.5, 0.5, width)
y_vals = np.linspace(-0.5, 0.5, height)
grid = np.meshgrid(x_vals, y_vals)
v_fn = np.vectorize(exp_z_inv)
return v_fn(*grid).astype(np.float)
fig = vp.Fig(size=(800, 600), show=False)
plot = fig[0, 0]
plot.bgcolor = "#efefef"
img = gen_image(500, 500)
plot.image(img, cmap="hsl")
plot.camera.set_range((100, 400), (100, 400))
positions = ["top", "bottom", "left", "right"]
for position in positions:
plot.colorbar(position=position,
label="argument of e^(1/z)",
clim=("0°", "180°"),
cmap="hsl",
border_width=1,
border_color="#aeaeae")
if __name__ == '__main__':
fig.show(run=True)
|
