Question

我试图绘制一个球在垂直振荡的表面上弹跳。我认为代码（与变量有关）正在运行，但是由于循环，数字不会更新以反映变量的变化。我希望这不是一个模糊的问题，但为什么他们不更新？

谢谢！

import numpy as np
import matplotlib.pyplot as plt


    h_b=np.array([10])#initial ball height
    g=-9.8 #gravity acceleration
    #v_b=np.array([0]) 
    v_b=np.array([0.0001])#initial ball velocity
    dt=0.1 #time-step

    a=3 #amplitude of surface's movement
    f=0.2 #frequency of the oscillations
    h_s=np.array([0]) #height of the oscillating surface 
    v_s=np.array([a*2*np.pi*f]) #velocity of the oscillating surface

    r=0.9 #coefficient of restitution

    x=np.linspace(0,20,20)

    t=np.array([0]) # setting up the time array


    fig= plt.figure()
    ax1=fig.add_subplot(1,2,1)
    ax2=fig.add_subplot(1,2,2)
    while np.abs(h_b[-1])>0.00001 and np.abs(v_b[-1])>0.00001 :
        while h_b[-1]>h_s[-1] :
            n=len(v_b)
            ax1.clear()
            ax1.set_xlim(0,20)
            ax1.set_ylim(-a,50)
            ax1.plot(10,h_b[n-1],'.',x,np.ones(20)*h_s[n-1])
            ax2.clear()
            ax2.plot(t,h_b)
            plt.show()
            v_b=np.append(v_b,v_b[n-1]+g*dt)
            h_b=np.append(h_b,h_b[n-1]+v_b[n-1]*dt)
            h_s=np.append(h_s,a*np.sin(2*np.pi*f*n*dt))
            v_s=np.append(v_s,a*2*np.pi*f*np.cos(2*np.pi*f*n*dt))
            t=np.append(t,n*dt)
            plt.pause(0.1)
        v_b[-1]=v_s[-2]-r*(v_b[-2]-v_s[-2])
        print('%f' % v_b[-1])
        h_b[-1]=h_s[-1]+0.0001

Answer 1

你不应该在循环中调用plt.show。它只能在脚本结束时调用一次。如果删除plt.show（），它将按预期工作。

import numpy as np
import matplotlib.pyplot as plt

plt.ion()
h_b=np.array([10])#initial ball height
g=-9.8 #gravity acceleration
#v_b=np.array([0]) 
v_b=np.array([0.0001])#initial ball velocity
dt=0.1 #time-step

a=3 #amplitude of surface's movement
f=0.2 #frequency of the oscillations
h_s=np.array([0]) #height of the oscillating surface 
v_s=np.array([a*2*np.pi*f]) #velocity of the oscillating surface

r=0.9 #coefficient of restitution

x=np.linspace(0,20,20)

t=np.array([0]) # setting up the time array


fig= plt.figure()
ax1=fig.add_subplot(1,2,1)
ax2=fig.add_subplot(1,2,2)
plt.draw()

while np.abs(h_b[-1])>0.00001 and np.abs(v_b[-1])>0.00001 :

    while h_b[-1]>h_s[-1] :
        n=len(v_b)
        ax1.clear()
        ax1.set_xlim(0,20)
        ax1.set_ylim(-a,50)
        ax1.plot(10,h_b[n-1],'.',x,np.ones(20)*h_s[n-1])
        ax2.clear()
        ax2.plot(t,h_b)

        v_b=np.append(v_b,v_b[n-1]+g*dt)
        h_b=np.append(h_b,h_b[n-1]+v_b[n-1]*dt)
        h_s=np.append(h_s,a*np.sin(2*np.pi*f*n*dt))
        v_s=np.append(v_s,a*2*np.pi*f*np.cos(2*np.pi*f*n*dt))
        t=np.append(t,n*dt)
        plt.pause(0.1)
    v_b[-1]=v_s[-2]-r*(v_b[-2]-v_s[-2])
    print('%f' % v_b[-1])
    h_b[-1]=h_s[-1]+0.0001

plt.ioff()
plt.show()

循环数字不更新

1 个答案: