u

src/yapsut/graphics.py

+import matplotlib.pyplot as plt
+import numpy as np
+
+class ImshowXT :
+   """ class to handle an improved version of plt.imshow """
+   def __init__(self,matr,x_values,y_values,**kwargs) :
+      """ matr = the matrix to show,
+            x_values = the values on the x axis,
+            y_values, the values on the y axis
+            kwarg = kwargs for plt.imshow
+      """
+      from matplotlib import pyplot as plt
+      self._im=plt.imshow(matr,**kwargs)
+      self._xv=x_values
+      self._yv=y_values
+   def set_xaxis(self,major_values,major_format) :
+      """ format the x axis for a give list of tick values and a formatting string """
+      from matplotlib import pyplot as plt
+      import numpy as np
+      self._mxv=major_values
+      self._mxf=major_format
+      #
+      v=self._xv
+      t=np.array([np.argmin(np.abs(k-v)) for k in major_values])
+      t1=np.arange(len(v))
+      #
+      plt.xticks(t,np.interp(t,t1,v))
+      tt=plt.gca().get_xticklabels()
+      for ik in range(len(tt))  :
+         o=float(tt[ik].get_text())
+         o1=major_format%o
+         tt[ik].set_text(o1)
+      plt.gca().set_xticklabels(tt)
+   def set_yaxis(self,major_values,major_format) :
+      """ format the y axis for a give list of tick values and a formatting string """
+      from matplotlib import pyplot as plt
+      import numpy as np
+      self._myv=major_values
+      self._myf=major_format
+      #
+      v=self._yv
+      t=np.array([np.argmin(np.abs(k-v)) for k in major_values])
+      t1=np.arange(len(v))
+      #
+      plt.yticks(t,np.interp(t,t1,v))
+      tt=plt.gca().get_yticklabels()
+      for ik in range(len(tt))  :
+         o=float(tt[ik].get_text())
+         o1=major_format%o
+         tt[ik].set_text(o1)
+      plt.gca().set_yticklabels(tt)
+   def colorbar(self,**karg) :
+      """ add a color bar """
+      from matplotlib import pyplot as plt
+      plt.colorbar(**karg)
+
 def odot(x,y,color=['yellow','red'],markersize=[30,10],scale=1.,zorder=None,mew=[4,0]) :
    """ draw an odot"""
    from matplotlib import pyplot as plt