surf example
"""
from pyvectorized import dom2vec, surf, \
newax, axis, axeq, hold, grid, gridhold
def myfun(x):
return x[0,:]**2 +x[1,:]**3
domain = [0, 1,0, 2]
resolution = [20, 30]
q = dom2vec(domain, resolution)
f = myfun(q)
ax, fig = newax(1, dim=[3])
surf(q, f, resolution, ax=ax)
axis(ax, domain +[0, 10])
axeq(ax)
"""manage multiple axes"""
ax2, fig = newax(1, dim=[3])
surf(q, -f, resolution, ax=ax2)
axs =[ax, ax2]
hold(axs)
grid(axs)
# shorthand for above
"""
plot 2d, 3d
"""
from pyvectorized import newax, dom2vec, \
plot, quiver, text, streamplot, axis
import numpy as np
x = np.arange(10)
y = np.square(x)
q2 = np.vstack([x, y])
t = np.arange(10)
q = np.vstack([t, np.sin(t), np.cos(t) ])
ax, fig = newax(2, dim=[2,3])
plot(q2, ax[0])
plot(q, ax[1])
"""
quiver 2d
"""
dom = [-10, 10, -10, 10]
res = [10, 10]
x = dom2vec(dom, res)
A = np.array([[1, -2],
[2, 1] ])
v = A.dot(x)
ax, fig = newax()
