def _update(self):
if self.x is not None and self.y is not None:
self.X, self.Y = num.meshgridt(self.x, self.y)
self.xx = self.X.flatten()
self.yy = self.Y.flatten()
self.XY = np.array([self.xx, self.yy]).T
elif self.X is not None and self.Y is not None:
self.x = self.X[:, 0]
self.xx = self.X.flatten()
self.y = self.Y[0, :]
self.yy = self.Y.flatten()
self.XY = np.array([self.xx, self.yy]).T
elif self.xx is not None and self.yy is not None:
self.x = self._unique(self.xx)
self.y = self._unique(self.yy)
self.X, self.Y = num.meshgridt(self.x, self.y)
self.XY = np.array([self.xx, self.yy]).T
elif self.XY is not None:
self.xx = self.XY[:, 0]
self.yy = self.XY[:, 1]
self.x = self._unique(self.xx)
self.y = self._unique(self.yy)
self.X, self.Y = num.meshgridt(self.x, self.y)
else:
raise Exception("cannot determine x and y from input")
# by now, we have all forms of x and y: x,y; xx,yy; X,Y; XY
self.nx = len(self.x)
self.ny = len(self.y)
# Z is optional
if self.Z is None:
if self.zz is not None:
self.Z = self.zz.reshape(len(self.x), len(self.y))
else:
self.zz = self.Z.flatten()
def _update(self):
if [self.x,self.y] != [None]*2:
self.X,self.Y = num.meshgridt(self.x, self.y)
self.xx = self.X.flatten()
self.yy = self.Y.flatten()
self.XY = np.array([self.xx, self.yy]).T
elif [self.X,self.Y] != [None]*2:
self.x = self.X[:,0]
self.xx = self.X.flatten()
self.y = self.Y[0,:]
self.yy = self.Y.flatten()
self.XY = np.array([self.xx, self.yy]).T
elif [self.xx,self.yy] != [None]*2:
self.x = self._unique(self.xx)
self.y = self._unique(self.yy)
self.X,self.Y = num.meshgridt(self.x, self.y)
self.XY = np.array([self.xx, self.yy]).T
elif self.XY is not None:
self.xx = self.XY[:,0]
self.yy = self.XY[:,1]
self.x = self._unique(self.xx)
self.y = self._unique(self.yy)
self.X,self.Y = num.meshgridt(self.x, self.y)
else:
raise StandardError("cannot determine x and y from input")
# by now, we have all forms of x and y: x,y; xx,yy; X,Y; XY
self.nx = len(self.x)
self.ny = len(self.y)
# Z is optional
if self.Z is None:
if self.zz is not None:
self.Z = self.zz.reshape(len(self.x), len(self.y))
else:
self.zz = self.Z.flatten()
def __init__(self, xlim, ylim, nx, ny, mode):
self.xlim = xlim
self.ylim = ylim
self.nx = nx
self.ny = ny
self.xg, self.yg = self.get_xy_grid()
self.XG, self.YG = num.meshgridt(self.xg, self.yg)
self.XY = self.make_XY(mode)
def get_2d_testdata(n=20):
"""2D sin + cos data.
Returns
-------
ret : :class:`Data2D`
"""
x = np.linspace(-5, 5, n)
X, Y = num.meshgridt(x, x)
Z = np.sin(X) + np.cos(Y)
return Data2D(X=X, Y=Y, Z=Z)
def meshgridt(*args, **kwds):
## warnings.simplefilter('always')
warnings.warn("mpl.meshgridt is deprecated, use num.meshgridt",
DeprecationWarning)
return num.meshgridt(*args, **kwds)
def meshgridt(*args, **kwds):
warnings.simplefilter('always')
warnings.warn("mpl.meshgridt is deprecated, use num.meshgridt",
DeprecationWarning)
return num.meshgridt(*args, **kwds)
def get_2d_testdata():
x = np.linspace(-5,5,20)
y = np.linspace(-5,5,20)
X,Y = num.meshgridt(x,y)
Z = np.sin(X) + np.cos(Y)
return Data2D(X=X, Y=Y, Z=Z)
