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

Question

我有一个np.array h_data包含九个相同长度的np.array。我想通过使用变量latitude（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])

我想要的是什么。但看起来很难看。