In [1]:
import numpy as np

In [2]:
import matplotlib.pyplot as plt
%matplotlib inline

In [3]:
def quadrado(x):
    return x**2

In [4]:
def cubo(x):
    return x**3

In [5]:
def baskara(a, b, c):
    x1 = (-b+np.sqrt(b**2-4*a*c))/(2*a)
    x2 = (-b-np.sqrt(b**2-4*a*c))/(2*a)
    return (x1, x2)

Teste das Funções


In [6]:
assert quadrado(2) == 4
assert quadrado(3) == 9
assert quadrado(5) == 25
assert quadrado(1) == 1
assert quadrado(12) == 144

In [7]:
assert cubo(3) == 27
assert cubo(1) == 1
assert cubo(2) == 8
assert cubo(5) == 125
assert cubo(4) == 64

In [8]:
assert baskara(2, -5, -3) == (3, -0.5)
assert baskara(2, -5, 3) == (1.5, 1)
assert baskara(1,0, -4) == (2, -2)
assert baskara(1, -4, -9) == (2+np.sqrt(13), 2-np.sqrt(13))
assert baskara(-1, 0, 9) == (-3, 3)

In [9]:
print ('done')


done

Gráficos


In [10]:
x = np.linspace(0, 99)
y = quadrado(x)
plt.plot(x, y)
plt.grid()
plt.show()



In [11]:
x = np.linspace(0, 99)
y = cubo(x)
plt.plot(x, y)
plt.show()



In [12]:
def func1(X):
    return 2*x**2-5*x-3

In [13]:
x = np.linspace(-10, 10)
y = func1(x)
x1, x2 = baskara(2, -5, -3)
plt.plot(x, y)
plt.plot(x1, 0, "o")
plt.plot(x2, 0, "o")
plt.grid()
plt.show()