In [1]:
from sympy.plotting import plot, plot_parametric, plot3d, plot3d_parametric_line, plot3d_parametric_surface
In [2]:
p = plot(sin(x))
If the line_color aesthetic is a function of arity 1 then the coloring is a function of the x value of a point.
In [3]:
p[0].line_color = lambda a : a
p.show()
If the arity is 2 then the coloring is a function of both coordinates.
In [4]:
p[0].line_color = lambda a, b : b
p.show()
In [5]:
p = plot_parametric(x*sin(x), x*cos(x), (x, 0, 10))
If the arity is 1 the coloring depends on the parameter.
In [6]:
p[0].line_color = lambda a : a
p.show()
For arity 2 the coloring depends on coordinates.
In [7]:
p[0].line_color = lambda a, b : a
p.show()
In [8]:
p[0].line_color = lambda a, b : b
p.show()
Arity 1 - the first parameter. Arity 2 or 3 - the first two coordinates or all coordinates.
In [9]:
p = plot3d_parametric_line(sin(x)+0.1*sin(x)*cos(7*x),
cos(x)+0.1*cos(x)*cos(7*x),
0.1*sin(7*x),
(x, 0, 2*pi))
In [10]:
p[0].line_color = lambda a : sin(4*a)
p.show()
In [11]:
p[0].line_color = lambda a, b : b
p.show()
In [12]:
p[0].line_color = lambda a, b, c : c
p.show()
In [14]:
p = plot3d(sin(x)*y, (x, 0, 6*pi), (y, -5, 5))
Arity 1, 2 or 3 for first, the two first or all coordinates.
In [15]:
p[0].surface_color = lambda a : a
p.show()
In [16]:
p[0].surface_color = lambda a, b : b
p.show()
In [17]:
p[0].surface_color = lambda a, b, c : c
p.show()
In [18]:
p[0].surface_color = lambda a, b, c : sqrt((a-3*pi)**2+b**2)
p.show()
Arity 1 or 2 - first or both parameters.
In [19]:
p = plot3d_parametric_surface(x*cos(4*y), x*sin(4*y), y,
(x, -1, 1), (y, -1, 1))
In [20]:
p[0].surface_color = lambda a : a
p.show()
In [21]:
p[0].surface_color = lambda a, b : a*b
p.show()
Arrity of 3 will color by coordinates.
In [22]:
p[0].surface_color = lambda a, b, c : sqrt(a**2+b**2+c**2)
p.show()