数组作为函数

Question

我定义了一个公式，我希望将现有的数组数据作为此函数的输入，并将结果返回到数组中。

代码

x1=np.array(x1)#     <---- existing input data

x_test=[0,5,10,15]    <---- just some test data
x_test=np.array(x_test)

def P_0(x):    
    return 295*(math.cos((0.9952*x+2.25952)*math.pi/180))**2-(295*(math.cos((0.9952*x+1.74)*math.pi/180))**2) 
results=P_0(x_test)

我收到错误：

TypeError: Required argument 'shape' (pos 1) not found

Answer 1

尝试更改：

def P_0(x):    
    return 295*(np.cos((0.9952*x+2.25952)*np.pi/180))**2-(295*(np.cos((0.9952*x+1.74)*np.pi/180))**2) 

我刚刚将math.pi替换为np.pi，将math.sin替换为np.sin，将math.cos替换为np.cos

如果在执行numpy数组操作时使用numpy中的函数，总会更好