How do I plot multiple functions in matplotlib?

Question

I have the following code:

def f(x):
    return x

def g(x):
    return x*x

from math import sqrt
def h(x):
    return sqrt(x)

def i(x):
    return -x

def j(x):
    return -x*x

def k(x):
    return -sqrt(x)

functions = [f, g, h, i, j, k]

And now I'm trying to plot these functions.

I tried

plt.plot(f(x), g(x), h(x))

but I get the following error:

TypeError: only length-1 arrays can be converted to Python scalars

I figure this is because I'm using the square root which has two solutions. But really, I'm trying to do something like:

plt.plot(*functions)

Any advice?

Please, provide a verifiable example even it's bugged : it shows what you tried so far. — IMCoins
– IMCoins, Commented Dec 26, 2017 at 15:08

unutbu · Accepted Answer · 2017-12-26 15:13:29Z

3

math.sqrt accepts only scalar values. Use numpy.sqrt to compute the square root of each value in a list or NumPy array:

In [5]: math.sqrt(np.array([0,1]))
TypeError: only length-1 arrays can be converted to Python scalars

In [6]: np.sqrt(np.array([0,1]))
Out[6]: array([ 0.,  1.])

import numpy as np
import matplotlib.pyplot as plt

def f(x):
    return x

def g(x):
    return x*x

def h(x):
    return np.sqrt(x)

def i(x):
    return -x

def j(x):
    return -x*x

def k(x):
    return -np.sqrt(x)

x = np.linspace(0, 1, 100)
functions = [f, g, h, i, j, k]
for func in functions:
    plt.plot(func(x), label=func.__name__)
plt.legend(loc='best')
plt.show()

answered Dec 26, 2017 at 15:13

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

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

Collectives™ on Stack Overflow

How do I plot multiple functions in matplotlib?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related