python - 使用条件仅提取 np.array 的一部分

Question

我有一个 np.arrayh_data包含九个相同长度的单独 np.arrays。我想通过使用变量纬度（h_data 的第三个数组）来整理数据。条件有效，请参阅下面的代码行。当我尝试在我的 h_data 数组上使用我的条件时，我得到一个错误。我明白为什么会发生这个错误。cond_lat和的长度h_data不同。我该如何解决这个问题？

cond_lat = (h_data[2,:] >= -25.901) & (h_data[2,:] <= -15.901)
new_h_data = h_data[cond_lat]

追溯

文件“read_NetCDF.py”，第 153 行

new_h_data = h_data[cond_lat]

索引错误：索引 9 超出轴 0 的范围，大小为 9

Answer 1

我找到了解决方案。

new_h_data0 = h_data[0,:][cond_lat]
new_h_data1 = h_data[1,:][cond_lat]
new_h_data2 = h_data[2,:][cond_lat]
new_h_data3 = h_data[6,:][cond_lat]

new_h_data = np.array([new_h_data0, new_h_data1, new_h_data2, new_h_data3])

做我想做的。但是长得丑。