Включает в себя функции нескольких условий

Question

Включает в себя функции нескольких условий

у меня есть массив расстояний, называемых dists. Я хочу выбрать дисты, которые находятся между двумя значениями. Для этого я написал следующую строку кода:

 dists[(np.where(dists >= r)) and (np.where(dists <= r + dr))]

однако это выбирает только для условия

 (np.where(dists <= r + dr))

Если я выполнить команды последовательно, используя временную переменную, она отлично работает. Почему приведенный выше код не работает, и как заставить его работать?

Ура

5 75

python numpy

5 ответов:

Так как принятый ответ очень хорошо объяснил проблему. вы также можете использовать numpy логические функции что больше подходит здесь для нескольких условий :
np.where(np.logical_and(np.greater_equal(dists,r),np.greater_equal(dists,r + dr)))

25

попробуй:

np.intersect1d(np.where(dists >= r)[0],np.where(dists <= r + dr)[0])

2

мне нравится использовать np.vectorize для таких задач. Рассмотрим следующее:
>>> # function which returns True when constraints are satisfied.
>>> func = lambda d: d >= r and d<= (r+dr) 
>>>
>>> # Apply constraints element-wise to the dists array.
>>> result = np.vectorize(func)(dists) 
>>>
>>> result = np.where(result) # Get output.
вы также можете использовать np.argwhere вместо np.where для четкого вывода. Но это ваш вызов:)

надеюсь, что это помогает.

2

Я разработал этот простой пример
import numpy as np

ar = np.array([3,4,5,14,2,4,3,7])

print [X for X in list(ar) if (X >= 3 and X <= 6)]

>>> 
[3, 4, 5, 4, 3]

0

askewchan · Accepted Answer · 2013-05-04 21:12:52

лучший способ в ваш частный случай было бы просто изменить ваши два критерия на один критерий:
dists[abs(dists - r - dr/2.) <= dr/2.]
он создает только один логический массив, и, на мой взгляд, его легче читать, потому что он говорит:и dist внутри dr или r? (хотя я бы переопределил r чтобы быть центром вашего региона интереса вместо начала, так что r = r + dr/2.) но это не отвечает вашим вопрос.

ответ на ваш вопрос:
Вам на самом деле не нужно where если вы просто пытаетесь отфильтровать элементы dists это не соответствует вашим критериям:
dists[(dists >= r) & (dists <= r+dr)]
потому что & даст вам элементарно and (скобки не нужны).

или, если вы хотите использовать where по какой-то причине, вы можете делать:
 dists[(np.where((dists >= r) & (dists <= r + dr)))]
почему:
Причина этого не работает, потому что np.where возвращает список индексов, а не булевский массив. Вы пытаетесь получить and между двумя списками чисел, которые, конечно, не имеют True/False значения, которые вы ожидаете. Если a и b как True значения, то a and b возвращает b. Так что говорю что-то вроде [0,1,2] and [2,3,4] только [2,3,4]. Вот он в действие:
In [230]: dists = np.arange(0,10,.5)
In [231]: r = 5
In [232]: dr = 1

In [233]: np.where(dists >= r)
Out[233]: (array([10, 11, 12, 13, 14, 15, 16, 17, 18, 19]),)

In [234]: np.where(dists <= r+dr)
Out[234]: (array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12]),)

In [235]: np.where(dists >= r) and np.where(dists <= r+dr)
Out[235]: (array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12]),)
то, что вы ожидали сравнить, было просто логическим массивом, например
In [236]: dists >= r
Out[236]: 
array([False, False, False, False, False, False, False, False, False,
       False,  True,  True,  True,  True,  True,  True,  True,  True,
        True,  True], dtype=bool)

In [237]: dists <= r + dr
Out[237]: 
array([ True,  True,  True,  True,  True,  True,  True,  True,  True,
        True,  True,  True,  True, False, False, False, False, False,
       False, False], dtype=bool)

In [238]: (dists >= r) & (dists <= r + dr)
Out[238]: 
array([False, False, False, False, False, False, False, False, False,
       False,  True,  True,  True, False, False, False, False, False,
       False, False], dtype=bool)
теперь вы можете звонить np.where на объединенном логическом массиве:
In [239]: np.where((dists >= r) & (dists <= r + dr))
Out[239]: (array([10, 11, 12]),)

In [240]: dists[np.where((dists >= r) & (dists <= r + dr))]
Out[240]: array([ 5. ,  5.5,  6. ])
или просто индексировать исходный массив с помощью логического массива с помощью необычные индексации
In [241]: dists[(dists >= r) & (dists <= r + dr)]
Out[241]: array([ 5. ,  5.5,  6. ])