Python 在5D掩码numpy数组上循环

我是python的初学者，我在5D数组中遇到了一个问题，即在循环进行一些计算时，要保持最后2个维度的大小不变，因为它们表示我将根据此计算绘制的地图的纬度和经度。我要应用的计算是在一个名为（line）的函数中，因为我在循环中调用这个函数来对（wind_data）5D数组进行计算。但是我有一个错误（ValueError：使用序列设置数组元素）

扫掠面积=5026
空气密度=1.225
def管路（v）：
prob=np.空（shape=v.shape）
idx_low=np，其中（v=4）和（v=15）和（v=25）
m=0.5*扫掠面积*空气密度*0.29*v**3*10**-3
概率[idx_低]=0
prob[idx_mid]=m[（v>=4）和（v