python - 在 Python for 循环中创建多个图

Question

我想在单个图中为不同的 Pa 值创建一个（T/Tmax vs R/R0）图，如下所示。我已经编写了将值附加到列表的代码，但是 (T/Tmax vs R/R0) 的所有值都附加在单个列表中，这并没有给出好的图。我该怎么做才能有这样的情节？另外，如何从第 1 列是 T/Tmax 列表和第 2、3、4 列的循环中的数据制作 excel 表......是不同 pa 的相应 R/R0 值？

KLMDAT1 = []

KLMDAT2 = []

for j in range(z):

    pa[j] = 120000-10000*j

    i = 0
    R = R0
    q = 0
    T = 0

    while (T<Tmax):

        k1 = KLM_RKM(i*dT,R,q,pa[j])
        k2 = KLM_RKM((i+0.5)*dT,R,q+0.5*dT*k1,pa[j])
        k3 = KLM_RKM((i+0.5)*dT,R,q+0.5*dT*k2,pa[j])
        k4 = KLM_RKM((i+1)*dT,R,q+dT*k3,pa[j])

        q = q +1/6.0*dT*(k1+2*k2+2*k3+k4)
        R = R+dT*q

        if(R>0):

            KLMDAT1.append(T / Tmax)
            KLMDAT2.append(R / R0)

        if(R>Rmax):
                Rmax = R

        if (abs(q)>c or R < 0):
                    break
        T=T+dT
        i = i+1

wb.save('KLM.xlsx')

np.savetxt('KLM.csv',[KLMDAT1, KLMDAT2])

plt.plot(KLMDAT1, KLMDAT2)

plt.show()

标签： pythonpython-3.xmatplotlib

Answer 1

你画错了。您的第一个变量需要是 T/Tmax。所以初始化一个空的 T 列表，将 T 值附加到它，将它除以 Tmax，然后绘制两次：第一次KLMDAT1，然后KLMDAT2。以下伪代码解释它

KLMDAT1 = []
KLMDAT2 = []
T_list = [] # <--- Initialize T list here

for j in range(z):
    ...

    while (T<Tmax):
        ...

        T=T+dT
        T_list.append(T) # <--- Append T here
        i = i+1

# ... rest of the code

plt.plot(np.array(T_list)/Tmax, KLMDAT1) # <--- Changed here
plt.plot(np.array(T_list)/Tmax, KLMDAT2) # <--- Changed here

plt.show()