SymPy 的彩色绘图答案

【问题标题】：Multicolored Plot from SymPySymPy 的彩色绘图
【发布时间】：2016-11-22 21:31:00
【问题描述】：

我正在尝试在 SymPy 中绘制函数 x^2 的图，并希望用线 x=3、y=9 覆盖它，然后还围绕某些点画一个圆。我已经用代码完成了上述所有操作

import sympy as sp
x = sp.Symbol('x')

# first plot the function
sp.plotting.plot_parametric( (x, x**2, (x,0,4)),
    # then the straight lines 
    (3, x, (x,0,16)), (x, 9, (x,0,4)), 
    # then the circle
    (.1*sp.cos(x)+3.1, .1*sp.sin(x)+(3.1**2), (x,0,6.5)) )

但是，所有的曲线都是蓝色的，我想让每条曲线都变成不同的颜色。当我有几条曲线时，查看文档并没有告诉我如何做到这一点，只有当我有一条曲线时，我在几个地方玩弄line_color='red' 并没有让我到任何地方。有人知道适当的方法或好的 hack 吗？

【问题讨论】：

标签： python plot sympy

【解决方案1】：

您已经绘制了三个函数，因此您有三个图，以通常的方式编号。您可以通过以下方式单独修改它们的所谓美学。不要阅读关于这一点的文档，其中说 line_color 应该是一个返回浮点数的函数。如果你这样做，你可能会像我刚刚在一个黑洞里那样花一个小时。

>>> from sympy import *
>>> var('x')
>>> aPlot = plotting.plot_parametric( (x, x**2, (x,0,4)), (3, x, (x,0,16)), (x, 9, (x,0,4)), (.1*cos(x)+3.1, .1*sin(x)+(3.1**2), (x,0,6.5)) )
>>> aPlot[0].line_color='r'
>>> aPlot[1].line_color='g'
>>> aPlot.show()

【讨论】：