Python function that plots functions

Question

I'm currently trying to define a function in python that plots functions.

I've already done this one:

def plota_f(x0, xf, n):
  xpoints = []
  ypoints = []
  for i in range(0,n):
      xpoints.append(x0+(xf-x0)/n *i)
      ypoints.append(np.sin(x0+(xf-x0)/n *i))
  plt.plot(xpoints, ypoints)
  plt.show()

that plots a Sin function from x0 to xf with n steps, but I would like to be able to add a parameter f

def plota_f(f,x0,xf,n):

so that I could call like

plota_f(x**2,0,10,100)

and plot a quadratic function or call

plota_f(np.sin(x),0,10,100)

and plot a sin function. Is there a simple way of doing that?

Edit: this is the code I got after the answer

def plota_f(f,x0,xf,n):
xpoints = []
ypoints = []
for i in range(0,n):
    xpoints.append(x0+(xf-x0)/n *i)
    x=x0+(xf-x0)/n *i
    ypoints.append(eval(f))
plt.plot(xpoints, ypoints)
plt.show()

Corralien · Accepted Answer · 2022-02-03 16:27:38Z

2

It's pretty easy with numpy:

from x0 to xf with n steps

This is the definition of np.linspace

be able to add a parameter f

Use a lambda function

Demo:

def plota_f(f, x0, xf, n):
    # Use with caution, eval can be dangerous
    xpoints = np.linspace(x0, xf, n, endpoint=True)
    ypoints = eval(f)
    plt.plot(xpoints, ypoints)
    plt.show()

plota_f('x**2', 0, 10, 100)

edited Feb 3, 2022 at 16:27

answered Feb 3, 2022 at 16:03

Corralien

121k8 gold badges43 silver badges68 bronze badges

Sign up to request clarification or add additional context in comments.

4 Comments

Mad Physicist Over a year ago

We have a similar answer, but I went off the deep end parametrizing the functions. Clearly this approach deserves a +1 :)

Rafael Over a year ago

is there I way I can only type the 'x**2' part on the parameter of the function and hide the 'lambda x:' part somewhere else?

Corralien Over a year ago

Yes if you use eval to evaluate f. Now you have to pass f as a string because like lambda you have to delayed the evaluation of function until x is computed

Rafael Over a year ago

It worked! I will post the code here as an edit to the question

Mad Physicist · Accepted Answer · 2022-02-03 16:13:18Z

You are over-complicating the problem. Here is how I would plot a sine function with frequency f over some interval, say from t0 to t1 with n steps:

t = np.linspace(t0, t1, n)
fn = np.sin(2 * np.pi * f * t)
plt.plot(t, fn)

When you separate out the domain vs function values, you can see a path forward. For example, you could define a function like this:

def my_sin(f):
    def func(t):
        return np.sin(2 * np.pi * f * t)
    return func

Now you can do something like this now:

def plot_func(func, t0, t1, n):
    t = np.linspace(t0, t1, n)
    plt.plot(t, func(t))

You just have to define functions that accept a single parameter t, which is an array of the points you want to evaluate at:

def my_quad(a, b, c):
    def func(t):
        return a * t**2 + b * t + c
    return func

plot_func(my_sin(10), -10, 10, 1000)
plot_func(my_quad(3, 0, 0), -5, 5, 100)

Collectives™ on Stack Overflow

Python function that plots functions

2 Answers 2

4 Comments

Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

4 Comments

Comments

Related