Python 3.6中Matplotlib用户输入公式绘图报错的解决咨询

你遇到的ValueError是因为**input()获取到的是字符串类型**，当你把这个字符串直接传给plt.plot()时，Matplotlib会误把它当成线条/标记的格式字符串（比如'ro'表示红色圆点标记），而不是用于计算y轴数值的表达式，自然会报错"Illegal format string"。

下面提供两种可行的方案，你可以根据场景选择：

方案1：使用eval()快速实现（适合可信输入场景） #

eval()可以将字符串形式的Python表达式直接执行计算，刚好能把用户输入的公式转换成对应的y值数组。

修改后的代码：

import numpy as np
import matplotlib.pyplot as plt

def graph(x_range):
    x = np.array(x_range)
    # 获取用户输入的公式表达式
    formula = input("请输入关于x的公式：")
    # 用eval计算出y的数值数组
    y = eval(formula)
    plt.plot(x, y)
    plt.show()

graph(range(-10, 11))

测试时输入2*x，就能正常生成图像了。

⚠️ 注意：eval()会执行任何传入的字符串代码，如果是接收不可信用户的输入（比如公开网站的用户输入），存在代码注入风险，这种场景下不要用这个方法。

方案2：使用SymPy库安全解析数学表达式（推荐用于不可信输入场景） #

如果需要处理不可信的用户输入，或者要支持更复杂的数学表达式（比如sin(x)、sqrt(x)），可以用专门的符号数学库SymPy，它只会解析数学表达式，不会执行任意Python代码，安全性更高。

步骤1：安装SymPy #

先在终端执行安装命令：

pip install sympy

步骤2：修改后的代码 #

import numpy as np
import matplotlib.pyplot as plt
from sympy import symbols, parse_expr

def graph(x_range):
    x_vals = np.array(x_range)
    # 定义符号变量x
    x = symbols('x')
    # 获取并解析用户输入的数学表达式
    expr_str = input("请输入关于x的公式：")
    expr = parse_expr(expr_str)
    # 将符号表达式转换为可计算数值的函数
    calc_y = lambda val: expr.subs(x, val)
    # 计算所有x对应的y值（转换为float类型）
    y = np.array([calc_y(val) for val in x_vals], dtype=np.float64)
    
    plt.plot(x_vals, y)
    plt.show()

graph(range(-10, 11))

这个方案不仅支持2*x这类简单表达式，还能处理np.sin(x)、x**2 + 3*x + 1等复杂公式，同时避免了代码注入的风险。

内容的提问来源于stack exchange，提问作者HF1