Состояние на Numpy массивы

Question

У меня есть два массива с одинаковым числом элементов

X = [1,2,3,4,5,6,7,8,9] 
Y = [10,4,3,7,7,3,1,8,98] 

Я хотел бы, чтобы элементы X и Y такие, как 2<X<7. Как я могу сделать?

Хорошо это хорошо работает с

Y = Y[np.logical_and(X>2, X<5)] 
X = X[np.logical_and(X>2, X<5)] 

Спасибо большое!

Answer 1

Вы можете использовать numpy.logical_and:

>>> X = np.array([1,2,3,4,5,6,7,8,9]) 
>>> X[np.logical_and(X>2, X<7)] 
array([3, 4, 5, 6]) 

Answer 2

вы можете использовать цикл и if, а также вы можете использовать set() для держать деферент индексы:

>>> X = [1,2,3,4,5,6,7,8,9] 
>>> Y = [10,4,3,7,7,3,1,8,98] 
>>> X=[i for i in X if 2<i<7] 
>>> Y=[i for i in Y if 2<i<7] 
>>> X 
[3, 4, 5, 6] 
>>> Y 
[4, 3, 3] 
>>> set(Y) 
set([3, 4])