基于数值选择数组元素的Pythonic方法？

Question

我正在寻找pythonic（单线）方法从数组中提取一系列值 下面是一些示例代码，它将从x，y数据中提取数据元素> 2和< 8，并将put他们成为一个新的阵列。有没有办法在一条线上完成这一点？下面的代码工作，但似乎比它需要的更糟糕。 （注意：实际上，我在我的应用程序彩车工作）

import numpy as np 

x0 = np.array([0,3,9,8,3,4,5]) 
y0 = np.array([2,3,5,7,8,1,0]) 

x1 = x0[x0>2] 
y1 = y0[x0>2] 
x2 = x1[x1<8] 
y2 = y1[x1<8] 

print x2, y2 

这将打印

[3 3 4 5] [3 8 1 0] 

（b）部分的问题将是提取值说1 < x < 3和7 < x < 9以及其相应的值为y。

Answer 1

你可以连续使用&为逐元素logical and和|为逐元素logical or，使得条件2 < x0和x0 < 8变得

mask = (2 < x0) & (x0 < 8) 

---

例如布尔数组，

import numpy as np 

x0 = np.array([0,3,9,8,3,4,5]) 
y0 = np.array([2,3,5,7,8,1,0]) 

mask = (2 < x0) & (x0 < 8) 
x2 = x0[mask] 
y2 = y0[mask]  
print(x2, y2) 
# (array([3, 3, 4, 5]), array([3, 8, 1, 0])) 

mask2 = ((1 < x0) & (x0 < 3)) | ((7 < x0) & (x0 < 9))  
x3 = x0[mask2] 
y3 = y0[mask2] 
print(x3, y3) 
# (array([8]), array([7])) 

Answer 2

import numpy as np 

x0 = np.array([0,3,9,8,3,4,5]) 
y0 = np.array([2,3,5,7,8,1,0]) 
list(zip(*[(x,y) for x, y in zip(x0, y0) if 1<=x<=3 or 7<=x<=9])) 

# [(3, 9, 8, 3), (3, 5, 7, 8)]