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
|
"""Superimpose histograms and curves"""
import numpy as np
from vedo.pyplot import histogram, plot
from vedo import settings
settings.default_font = "Bongas"
settings.remember_last_figure_format = True
mu, sigma, n, bins = 100.0, 15.0, 600, 50
samples = np.random.normal(loc=mu, scale=sigma, size=n)
x = np.linspace(min(samples), max(samples), num=50)
y = 1/(sigma*np.sqrt(2*np.pi)) * np.exp( -(x-mu)**2 /(2*sigma**2))
dy = 1/np.sqrt(n)
fig = histogram(
samples,
title=__doc__,
bins=bins,
density=True,
c='cyan3',
aspect=9/7,
label="gaussian",
)
fig += plot(x, y, "-", lc='orange5', label="some fit")
fig += plot(x, y*(1+dy), "--", c='orange5', lw=2)
fig += plot(x, y*(1-dy), "--", c='orange5', lw=2)
fig.add_legend()
fig.show(size=(800,700), zoom="tight").close()
|