有没有办法可以在代码中使用np.array

时间:2015-06-01 09:23:39

标签: python arrays optimization np

我想知道是否有一种方法可以将此代码转换为np.array代码。然后将其添加到link。我想在球发射时增加一个角度。

import numpy as np
import scipy as sp
from scipy.integrate import ode
import matplotlib.pylab as pl
import matplotlib.pyplot as plt
import math
from matplotlib import*
from matplotlib.pyplot import *
from __future__ import division
import math

 def projectile_xy(initPos,g,initVel):
    data_xy = []
    initTime = 0.0
    while True:

现在计算高度y

        y = initPos + (initTime * initVel * math.sin(theta)) - (g * initTime * initTime)/2

弹丸已达到地面水平

        if y < 0:
            break

计算距离x

        x = initVel * math.cos(theta) * initTime

将(x,y)元组附加到列表

        data_xy.append((x, y))

以0.1秒的增量使用时间

        initTime += 0.1
    return data_xy

g = 9.8
#h = float(raw_input("Enter the height "))
initPos = float(raw_input("Enter the height "))
der = float(raw_input("Enter the angle "))
#v = float(raw_input("Enter velocity ")) 
initVel = float(raw_input("Enter velocity ")) 


theta = math.radians(der)  # radians
data_der = projectile_xy(initPos,g,initVel)

找到最大高度......

point_height_max = max(data_der, key = lambda q: q[1])
xm, ym = point_height_max
x_max = max(data_der)[0]

print('''
Projectile Motion ...
Using a firing angle of {} degrees
and a muzzle velocity of {} meters/second
the maximum height is {:0.1f} meters
at a distance of {:0.1f} meters'''.format(der, initVel, ym, xm))
print "maximum distance" ,(x_max)

输入高度1 输入角度45 输入速度30

Projectile Motion ...
Using a firing angle of 45.0 degrees
and a muzzle velocity of 30.0 meters/second
the maximum height is 24.0 meters
at a distance of 46.7 meters
maximum distance 91.2167747731

1 个答案:

答案 0 :(得分:1)

您可以遵循以下方法:

import numpy as np

linear_vel = 20
ang = np.pi/3
y=10
x=12
g=9.8
y_vel =  linear_vel*np.cos(ang)
x_vel =  linear_vel*np.sin(ang)


t = (y_vel+np.sqrt(y_vel**2+2*g*y))/g   #time when projectile hits               ground
n= 20 #number of instances of time you want

time_values = np.linspace(0,int(t),n)
axes = np.zeros((2,n))

for i in range(0,n):
    axes[0,i]=x_vel*time_values[i]
    axes[1,i]=y_vel*time_values[i]-0.5*g*time_values[i]**2



#print time_values
#print axes