import numpy as np
import matplotlib.pyplot as plt
dt=1e-5
dx=1e-2
a=0.5
alpha=a*dt/((dx)**2)
T0=10

Temps=np.arange(0,0.09+dt,dt)
position=np.arange(0,1+dx,dx)

T=np.zeros((len(Temps),len(position)))
for x in range(len(position)):
    T[0][x]=T0*np.sin(2*np.pi*position[x])
for n in range(len(Temps)-1):
    for j in range(1,len(position)-1):
            T[n+1][j]=T[n][j]+alpha*(T[n][j+1]-2*T[n][j]+T[n][j-1])
            
t=(len(Temps)//9)
plt.plot(position,T[0])
plt.plot(position,T[t])
plt.plot(position,T[2*t])
plt.plot(position,T[3*t])
plt.plot(position,T[4*t])
plt.plot(position,T[5*t])
plt.plot(position,T[6*t])
plt.plot(position,T[7*t])
plt.plot(position,T[8*t])
plt.plot(position,T[9*t])