Plotting

Getting started

Scipp uses the plopp package (https://scipp.github.io/plopp) for visualizing data.

To install plopp, use:

pip install plopp

or

conda install -c conda-forge -c scipp plopp

Making plots interactive

Matplotlib makes static plots by default in Jupyter notebooks. To enable interactive plots, use

%matplotlib widget

at the start of your notebook (see here for more details on Matplotlib backends).

Scipp offers a number of different ways to plot data from a Variable, DataArray, Dataset or a DataGroup.

Plotting functionality is available in two different ways:

• using the plot() free function

• using the .plot() method on a Scipp object (variable, data array, dataset or datagroup)

The difference between the two possible plot functions is that the free function can accept more input types than just the Scipp objects. It can also plot raw NumPy arrays, as well as Python dicts of Scipp variables or data arrays. For Scipp objects, the produced plot will be the same with either approach: Internally, the .plot() method just forwards the Scipp object to the free function plot().

Consider two data arrays storing 1-D data:

import numpy as np

import scipp as sc

size = 50
rng = np.random.default_rng(seed=0)
x = sc.linspace('x', 0.0, 2.0, num=size, unit='m')
y = sc.linspace('y', 0.0, 1.0, num=5, unit='us')
temp1 = sc.array(dims=['x'], values=rng.random(size), unit='K')
temp1 += sc.linspace('x', 100, 105, num=size, unit='K')
da1 = sc.DataArray(temp1, coords={'x': x})
da1.name = 'temp1'  # Data array name is optional and will be used as a label
temp2 = sc.array(dims=['x'], values=rng.random(size), unit='K')
temp2.variances = temp2.values + 1
temp2 += sc.linspace('x', 99, 102, num=size, unit='K')
da2 = sc.DataArray(temp2, coords={'x': x})

The information in a data array or dataset is typically enough to create meaningful plots:

da1.plot()

In the plot above, the x dimension has an associated x coordinate and its values are used to label the ticks on the horizontal axis. The coordinate name and the unit (here: m) are used as a label for the horizontal axis. In a 1-D plot the unit of the data values (here K) labels the vertical axis.

Multiple data arrays can be plotted by passing a Python dict to the plot function. This example also illustrates how a data array with uncertainties results in a plot with error bars:

sc.plot({'temp1': da1, 'temp2': da2})

When the data arrays are part of a dataset, we can plot this directly:

ds = sc.Dataset({'temp1': da1, 'temp2': da2})
ds.plot()

Plotting slices or items of a dataset

The usual indexing and slicing can be used to create plots of slices of data, or plots of individual items from a dataset.

Plot a single entry of a dataset

ds['temp2'].plot()

Plot a slice range

ds['x', 10:30].plot()

2-D data

Creating a 2-D plot

2-D data arrays can be plotted directly, just like 1-D data:

size = 50
rng = np.random.default_rng(seed=0)
x = sc.linspace('x', 1.0, 3.0, num=size, unit='m')
time = sc.linspace('time', 1.0, 2.0, num=2 * size, unit='us')
temp = sc.array(dims=['x', 'time'], values=rng.random((size, 2 * size)), unit='K')
temp += sc.linspace('x', 100, 105, num=size, unit='K')
da = sc.DataArray(temp, coords={'x': x, 'time': time})
da.name = 'temperature'  # Data array name is optional and will be used as a label

This result in the following 2-D plot:

da.plot()

Plot a 1-D slice of 2-D data

When slicing without a range, the dimensionality reduces. This can be used to, e.g., plot a 1-D slice through 2-D data:

da['time', 4].plot()

Transpose axes

To control which dimensions are shown along which axes of the figure, we transpose the data before calling plot:

da.transpose().plot()

Logarithmic scale

Data can be plotted on a logarithmic scale on one or both axes. For the independent axis (the coordinate axis, i.e., the horizontal axis) this can be set using the scale option:

da1.plot(scale={'x': 'log'})

da.plot(scale={'x': 'log', 'time': 'log'})

Note the the keys in the scale dict are dimension labels and not “x” and “y” as Matplotlib would refer to its axes.

For the dependent axis (the data axis, i.e., vertical axis) use the norm option:

(100 * (da1 - da1.min())).plot(norm='log')

Further documentation

To view the full plotting documentation, go to the Plopp pages.