熱拡散の計算02 (フーリエ級数展開)

import matplotlib.pyplot as plt
import math

time_max=8
timestep=0.005
count=240
n_max=30
result=[]
for t_ in range(time_max):
    result = []
    t = t_ * timestep
    for x in range(count):
        y=0
        for n in range(1,n_max):
            a = 8 * math.sin(math.pi * n / 2) / ((math.pi ** 2) * (n ** 2))
            y += a*math.sin(math.pi*n*1.0/count*x)*math.exp(-1*(n*math.pi)**2*t)
        result.append(y)
    plt.plot(result)