Density¶
1# Plot density
2# This example creates a highly efficient and responsive density plot.
3
4# Generate some dummy data.
5N = int(1e6)
6
7xs = numpy.zeros(2 * N)
8ys = numpy.zeros(2 * N)
9
10xs[:N] = numpy.random.normal(loc=0.8, scale=0.2, size=N)
11xs[N:] = numpy.random.normal(loc=0.2, scale=0.2, size=N)
12
13ys[:N] = numpy.random.normal(loc=0.4, scale=0.2, size=N)
14ys[N:] = numpy.random.normal(loc=0.7, scale=0.25, size=N)
15
16xs = xs * Angstrom
17ys = ys * Angstrom
18
19# Create a model and apply some settings.
20model = Plot.PlotModel(Angstrom, Angstrom)
21model.framing().setTitle('Plot Density')
22model.xAxis().setLabel('x')
23model.yAxis().setLabel('y')
24
25# Change y-axis autoscale padding.
26model.xAxis().setAutoscalePadding(0, 0)
27model.yAxis().setAutoscalePadding(0, 0)
28
29# Create and add density plot.
30density = Plot.Density(xs, ys)
31density.setBroadening(1)
32density.setColorMode(Plot.COLOR_MODES.COLOR_MAP)
33density.setDpi(150)
34density.setLimits(0, 5)
35model.addItem(density)
36
37# Autoscale.
38model.setLimits('x', 0, 1)
39model.setLimits('y', 0, 1)
40
41# Show the plot for interactive editing.
42Plot.show(model)
43
44# Save the plot (can also be saved to svg, pdf, jpeg or hdf5).
45Plot.save(model, 'density.png')
References: Plot.Density, Plot.PlotModel, Plot.show, Plot.save.
↩ Go back to Plots