43 lines
766 B
Python
43 lines
766 B
Python
from scipy.integrate import odeint
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
import pylab as pl
|
|
import math
|
|
|
|
#plot f(x) vs. x
|
|
x = np.arange(-4.0,4.0,0.01)
|
|
fx = 4*x**2 - 16
|
|
zerox = 0.*x
|
|
|
|
#Create plot env
|
|
f = plt.figure()
|
|
plt.grid(True)
|
|
f.set_figwidth(8)
|
|
f.set_figheight(8)
|
|
|
|
#plot some stuff
|
|
plt.plot(x,fx,'g-',x,zerox,'b-')
|
|
|
|
plt.xlabel('x',fontsize = 15)
|
|
plt.ylabel('y', fontsize = 15)
|
|
plt.savefig("2024-09-09/ex1.png")
|
|
|
|
#Integrating stuff!
|
|
#Define the integration:
|
|
def dx_dt(xx,t):
|
|
return[4.*xx[0]**2 -16.]
|
|
|
|
f1 = plt.figure()
|
|
f.set_figwidth(12)
|
|
f.set_figheight(12)
|
|
|
|
# forward integration in time, so we need a time span
|
|
ts = np.linspace(0,3,1000)
|
|
ic = 1
|
|
xs = odeint(dx_dt, ic, ts)
|
|
print(xs)
|
|
|
|
plt.plot(ts,xs)
|
|
plt.savefig("2024-09-09/ex2.png")
|
|
plt.show()
|