I have working code using sympy.lambdify for a user input function string and then to plot the function. It fails if the function entered is just a constant number.
The code I am using is
import sympy as sy
import numpy as np
import matplotlib.pyplot as plt
x = sy.symbols('x')
func = 5
x_range = np.linspace(0, 10, 100)
F = sy.lambdify(x, func, "numpy")
y_range = F(x_range)
plt.plot(x_range, y_range)
So if func is a valid function like "2x" or "sin(x) - 7x**2", the code works fine.
However, if I enter a constant, say "5", I get a single number in y_range, not an array like x_range with all elements equal to the value of the entered constant. Therefore, I get a value error
ValueError: x and y must have same first dimension, but have shapes (200,) and (1,)
I have played with this for so long, I am beginning to feel like it is an error or inconsistency in sympy.
Does any one have a suggestion please?