如何修改Sympy绘制的函数与导数图像的颜色？

问题说明 #

你现有Sympy绘图代码，需要修改原函数和导数图像的颜色，同时原代码中缺失导数dfdx的定义，需先补全。

解决方案 #

通过访问Sympy绘图对象的子图列表，为每个图像单独设置line_color属性即可修改颜色。以下是完整代码：

import sympy as smp
from sympy.plotting import plot

x = smp.symbols('x', real=True)
# 定义原函数
f = x * smp.sin(x) * smp.exp(-x**2)
# 计算函数的导数
dfdx = smp.diff(f, x)

# 创建绘图对象，设置show=False暂不显示
p = plot(f, dfdx, (x, -3, 3), 
         title="函数及其导数在区间[-3,3]的图像", 
         show=False)

# 为原函数图像设置颜色（示例为蓝色）
p[0].line_color = 'blue'
# 为导数图像设置颜色（示例为红色）
p[1].line_color = 'red'

# 显示修改后的图像
p.show()

关键说明 #

• p[0]对应绘图时传入的第一个函数（原函数），p[1]对应第二个函数（导数）
• 颜色支持英文名称（如'green'、'orange'）或十六进制颜色码（如'#2ECC71'）

原生成图像：

内容的提问来源于stack exchange，提问作者José Luis Rodríguez Pirela