GeezerVison is showing github.com — fastplotlib/examples/line/inf_line_cmap_transform.py at streamplot · fastplotlib/fastplotlib

Expand file tree

inf_line_cmap_transform.py

Copy path

More file actions

33 lines (24 loc) · 920 Bytes

Copy path

33 lines (24 loc) · 920 Bytes

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

"""

Infinite Lines Colormap Transform

=================================

Use a ``cmap_transform`` to color infinite lines by an associated value rather than by their sequential

order. Here each line at an x-position is colored according to the sine value at that x-axis position.

"""

# test_example = true

# sphinx_gallery_pygfx_docs = 'screenshot'

import fastplotlib as fpl

import numpy as np

figure = fpl.Figure(size=(700, 560))

# evenly spaced vertical lines

positions = np.linspace(0, 6 * np.pi, 32)

# color each line by an associated value using the colormap transform

values = np.sin(positions)

figure[0, 0].add_inf_line(

positions, axis="x", cmap="plasma", cmap_transform=values, thickness=3

)

figure.show()

# NOTE: fpl.loop.run() should not be used for interactive sessions

# See the "JupyterLab and IPython" section in the user guide

if __name__ == "__main__":

print(__doc__)

fpl.loop.run()