spheres.beams.colorize

spheres.beams.colorize(z)

Converts complex values into colors: hue represents phase and brightness magnitude.

Adapted from https://stackoverflow.com/questions/17044052/mathplotlib-imshow-complex-2d-array.

Parameters

z (np.array) – Complex values.

Returns

c – Color values.

Return type

np.array