python - 从 3D 数组中删除与从 3D 数组中删除的行相对应的行

Question

我正在处理具有这种形状的雷达反射率数据，我的最终目标是绘制它，但在此之前我遇到了这个问题 -

(2500,50,200) #(scans, rays, altitude)

我还有三个 3D numpy 数组，分别对应于纬度、经度和高度，它们的形状与雷达反射率相同，即

 (2500,50,200)

大部分数据由零组成。所以我认为在绘制它们之前删除与零相对应的条目是有意义的。删除反射率数据中由零组成的条目后，我还需要返回并删除纬度​​、经度和海拔数组中的相应条目。

这是一个玩具示例的尝试

import numpy as np

arr=np.arange(27).reshape((3,3,3))

arrNZ = arr[np.nonzero(arr)]

print(arr.shape,arrNZ)

indx = np.where(arr == 0)

arr1 = np.arange(27).reshape((3,3,3))

n_arr1 = np.delete(arr1,indx)

print(n_arr1.shape)

但这似乎不起作用。该元素似乎并未被删除。我哪里出错了？

Answer 1

这是你想要的，还是我误解了你的问题？

import numpy as np

ref = np.array([1,0,3,4,0,6,7,8,9]).reshape(3,3) # reflectivity
lat = np.arange(9).reshape(3,3)                  # latitude
print("ref =\n", ref)
print("lat =\n", lat)
ref2 = ref[np.nonzero(ref)] # keep only non-zero
lat2 = lat[np.nonzero(ref)] # keep the items where 'ref' is non-zero
print("ref2 =", ref2)
print("lat2 =", lat2)

结果:

ref =
 [[1 0 3]
  [4 0 6]
  [7 8 9]]
lat =
 [[0 1 2]
  [3 4 5]
  [6 7 8]]
ref2 = [1 3 4 6 7 8 9]
lat2 = [0 2 3 5 6 7 8]