flappy ball

code

 #引入画图包和数学包
import matplotlib.pyplot as plt
import math

#定义函数体
#相关参数为：速度，角度，自旋，空气阻力相关，模拟步长数
def A_FLY_BALL(V0,A,W,S0,B2,LIM):

#使用这个来保存X,Y,Z的值
    X=[0 for x in range(0,LIM)]
    Y=[0 for x in range(0,LIM)]
    Z=[0 for x in range(0,LIM)]

#定义球的初始位置
    X[0]=0
    Y[0]=0
    Z[0]=0

#定义步长和加速度
    T=0.01
    g=9.8

#定义速度变量并给出初始值
    VX=V0*math.cos(A)
    VY=V0*math.sin(A)
    VZ=0
    V=V0
#进入循环
    for i in range(1,LIM):
        VX=VX-(B2*V*VX)*T
        VY=VY-g*T
        VZ=VZ-(S0*W*VX)*T
        V=(VX*VX+VY*VY+VZ*VZ)**0.5
        X[i]=X[i-1]+VX*T
        Y[i]=Y[i-1]+VY*T
        Z[i]=Z[i-1]+VZ*T

#绘制Y、Z和X的图像
    P1=plt.plot(X,Y)
    P2=plt.plot(X,Z)

#返回结果    
    return P1,P2