根据条件选择一些元素

Question

我有少量元素RA和DEC以及z。它们有点表示元素的x，y，z位置。

它们是存储在数组中的值。

max(z) = 3.0 ，这是我未在此处展示的第三个维度。

我想要做的是，根据特定条件删除下图中框内的元素，并在框外提供这些元素（同时 z<1.0 ）

所以从图片中的示例来看， 我需要黑盒子外的所有元素 以及第三个条件 < em> z<1.0

我尝试过并且无效的是以下：

data_z = contains all the z values
data_RA = contains all the RA values
data_DEC = contains all the DEC values

ra_lim = np.array([[40.0,43.0],[43.0,46.0],[46.0,50.0]])
dec_lim = np.array([[4.0,8.0],[0.0,4.0],[-2.0,0.0]])

new_z = data_z[(data_z<1.0) * ~(data_ra>ra_lim[0][0]) * ~(data_ra<ra_lim[0][1])
                     * ~(data_dec>dec_lim[0][0]) * ~(data_dec<dec_lim[0][1])]

new_RA = data_RA[(data_z<1.0) * ~(data_ra>ra_lim[0][0]) * ~(data_ra<ra_lim[0][1])
                         * ~(data_dec>dec_lim[0][0]) * ~(data_dec<dec_lim[0][1])]

new_DEC = data_DEC[(data_z<1.0) * ~(data_ra>ra_lim[0][0]) * ~(data_ra<ra_lim[0][1])
                         * ~(data_dec>dec_lim[0][0]) * ~(data_dec<dec_lim[0][1])]

所以基本上是

~(data_ra>ra_lim[0][0]) * ~(data_ra<ra_lim[0][1])
                         * ~(data_dec>dec_lim[0][0]) * ~(data_dec<dec_lim[0][1])]

应表示框， ~ 符号应该为我提供反向条件。即排除此框中的值并返回剩余的值。

但这似乎不起作用。 我如何使用第三个条件z<1.0 来获取框外的值？

Answer 1

这是一个有效的小例子：

x = np.array([1,5,7,9,4,5,6,8,9,0,3])
y = np.array([2,4,5,7,3,2,6,2,7,8,3])

假设您希望获得由（x <6，y <5）定义的框之外的所有元素。该框由（x <6）*（y <5）表示。如果你想要框外，你可以采用相反的方法：~((x < 6) * (y < 5))。这等于~(x < 6) + ~(y < 5)而不是~(x < 6) * ~(y < 5)（如代码中所写）。

mask = ~(x < 6) + ~(y < 5)
print x[mask]
print y[mask]
## [7 9 6 8 9 0]
## [5 7 6 2 7 8]

在你的情况下，面具应该是

mask = (data_z<1.0) * ~((data_ra>ra_lim[0][0]) * (data_ra<ra_lim[0][1])
                     * (data_dec>dec_lim[0][0]) * (data_dec<dec_lim[0][1]))