热方程-设置和绘图

Question

任务是：

给出热方程问题

我需要的情节

生成2D图，其中使用5和100模式（α= 0.5时，在t = 0、1、2、3、4处显示温度曲线）（换句话说，不是计算无限总和，仅求和前5或100计算）。

我是python的新手，所以我不知道该如何解决。 这是我到目前为止的代码

x = np.arange(0,4)
n = np.arange(1,6)
t = np.arange(0,6)
def fun(x,n,t):
  l = 0
  for i in x:
    for j in n:
        for k in t:
            x = i
            n = j
            t = k
            z = []
            alpha = 0.5
            a = 8/m.pi
            b = m.sin((2*n - 1)*x)
            c = (2*n - 1)**2
            d = m.exp(-(2*n - 1)**2 *alpha**2 *t)
            T = a*(b/c)*d
            l += T
    z.append(T)
  return z

这不会运行。 您能帮我生成所需的2D图吗？ 打印（FUN（X，N，T））

Answer 1

我发现您的代码有两个问题：

1. 你总是在内部循环中重置z
2. 尽管T是在内部循环中计算的，但z.append(T)的缩进将z.append(T)标记为在第一个循环中。

我认为这是您想要的：

def fun(x,n,t):
  l = 0
  z = []
  for i in x:
    for j in n:
      for k in t:
        x = i
        # ... insert your other calculations here...
        T = a*(b/c)*d
        l += T
        z.append(T)
  return z