def energy_evolution_plot(time, kinetic, potential, thermal, figname="energy_evolution.png"):
native_plot.subplot(211)
plot(time, kinetic, label='K')
plot(time, potential, label='U')
plot(time, thermal, label='Q')
plot(time, kinetic + potential + thermal, label='E')
xlabel('Time')
ylabel('Energy')
native_plot.legend(prop={'size':"x-small"}, loc=4)
native_plot.subplot(212)
plot(time, thermal, label='Q')
native_plot.savefig(figname)
native_plot.clf()
def energy_evolution_plot(time, kinetic, potential, thermal, figname="energy_evolution.png"):
native_plot.subplot(211)
plot(time, kinetic, label='K')
plot(time, potential, label='U')
plot(time, thermal, label='Q')
plot(time, kinetic + potential + thermal, label='E')
xlabel('Time')
ylabel('Energy')
native_plot.legend(prop={'size':"x-small"}, loc=4)
native_plot.subplot(212)
plot(time, thermal, label='Q')
native_plot.savefig(figname)
native_plot.clf()
Creates a number of plots using the AMUSE unit aware plot functions.
"""
from amuse.units import units, quantities
from amuse.plot import (
native_plot, plot, scatter, xlabel, ylabel, hist
)
import numpy as np
if __name__ == "__main__":
# latex_support()
x = np.pi/20.0 * (range(-10, 10) | units.m)
y1 = units.MSun.new_quantity(np.sin(x.number))
y2 = units.MSun.new_quantity(x.number)
native_plot.subplot(2, 2, 1)
plot(x, y2, label='model')
scatter(x, y1, label='data')
xlabel('x')
ylabel('mass [$M_\odot$]') # overrides auto unit!
native_plot.legend(loc=2)
x = range(50) | units.Myr
y1 = quantities.new_quantity(
np.sin(np.arange(0, 1.5, 0.03)), 1e50*units.erg)
y2 = -(1e43 | units.J) - y1
native_plot.subplot(2, 2, 2)
plot(x, y1, label='$E_\mathrm{kin}$')
plot(x, y2, label='$E_\mathrm{pot}$')
xlabel('t')
ylabel('E')
from amuse.units import units, quantities
from amuse.plot import (
native_plot, plot, scatter, xlabel, ylabel, hist
)
import numpy as np
if __name__ == "__main__":
# latex_support()
x = np.pi / 20.0 * (list(range(-10, 10)) | units.m)
y1 = units.MSun.new_quantity(np.sin(x.number))
y2 = units.MSun.new_quantity(x.number)
native_plot.subplot(2, 2, 1)
plot(x, y2, label='model')
scatter(x, y1, label='data')
xlabel('x')
ylabel('mass [$M_\odot$]') # overrides auto unit!
native_plot.legend(loc=2)
x = list(range(50)) | units.Myr
y1 = quantities.new_quantity(
np.sin(np.arange(0, 1.5, 0.03)), 1e50 * units.erg)
y2 = -(1e43 | units.J) - y1
native_plot.subplot(2, 2, 2)
plot(x, y1, label='$E_\mathrm{kin}$')
plot(x, y2, label='$E_\mathrm{pot}$')
xlabel('t')
ylabel('E')