mirror of
https://github.com/triqs/dft_tools
synced 2024-11-01 11:43:47 +01:00
21 lines
665 B
Python
21 lines
665 B
Python
import numpy
|
|
|
|
class myObject(object):
|
|
def _plot_(self, options):
|
|
PI = numpy.pi
|
|
xdata = numpy.arange(-PI,PI,0.1)
|
|
phi = options.pop('phi',0) # Crucial : removes the entry from the dict
|
|
ydata1 = numpy.cos(xdata + phi)
|
|
ydata2 = numpy.sin(xdata + phi)
|
|
return( [
|
|
{'type' : "XY", 'xdata': xdata, 'ydata': ydata1, 'label': r'$x \rightarrow \cos (x + \phi), \quad \phi = %s$'%phi},
|
|
{'type' : "XY", 'xdata': xdata, 'ydata': ydata2, 'label': r'$x \rightarrow \sin (x + \phi), \quad \phi = %s$'%phi}
|
|
] )
|
|
|
|
X = myObject()
|
|
|
|
from pytriqs.plot.mpl_interface import oplot
|
|
oplot(X,'-o')
|
|
oplot(X,'-x', phi = 0.3)
|
|
|