#!/usr/bin/env python
#
# This example shows how to use the Function1D object to analyze and plot results.
import numpy as np
import matplotlib.pyplot as plt
from abipy.core import Function1D
# Build mesh [0, 2pi] with 100 points.
mesh = np.linspace(0, 2*np.pi, num=100)
# Compute sine function.
sine = Function1D.from_func(np.sin, mesh)
# Call matplotlib to plot data.
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
# Plot sine.
sine.plot_ax(ax, label="sin(x)")
# Spline sine on the coarse mesh sx, and plot the data.
sx = np.linspace(0, 2*np.pi, num=25)
splsine = sine.spline(sx)
plt.plot(sx, splsine, "ro", label="splined sine")
# Compute the 1-st and 2-nd order derivatives
# with finite differences (5-point stencil).
for order in [1,2]:
der = sine.finite_diff(order=order)
#!/usr/bin/env python import wx import numpy as np from abipy.core import Function1D from abipy.gui.awx.func1dframe import Func1dPlotFrame app = wx.App() func1d = Function1D.from_func(np.sin, np.arange(1,100,1)) frame = Func1dPlotFrame(None, func1d) frame.Show() app.MainLoop()
