Python найти индексы списка данного условие

Question

У меня есть список, скажем:

list_A = [0,0,0,1.0,2.0,3.0,2.0,1.0,0,0,0] 

Я хотел бы найти минимальные и максимальные показатели этого списка, где list_A> 0.

т.е. в приведенном выше примере, это будет 3 и 7.

для других списков, которые возрастают монотонно, я использую np.searchsorted, как np.searchsorted(list,[0.5,1.0]) найти индексы, в которых список равен between 0.5 and 1.0 соответственно.

Но этот случай совсем другой, и np.searchsorted здесь не работает, или, может быть, он делает это так, как я этого не знаю!

Answer 1

Фильтр архивный список с его indixes и взять минимальное и максимальное:

>>> list_A = [0,0,0,1.0,2.0,3.0,2.0,1.0,0,0,0] 
>>> filtered_lst = [(x,y) for x,y in enumerate(list_A) if y > 0] 
>>> max(filtered_lst) 
(7, 1.0) 
>>> min(filtered_lst) 
(3, 1.0) 

Если вам просто нужно индекс, распаковывать возвращаемое значение:

>>> maX,_ = max(filtered_lst) 
>>> maX 
7 

Answer 2

Альтернативой было бы использовать next():

list_A = [0,0,0,1.0,2.0,3.0,2.0,1.0,0,0,0] 

print(next(idx for idx, item in enumerate(list_A) if item>0)) 
print(next(len(list_A)-1-idx for idx, item in enumerate(list_A[::-1]) if item>0)) 

Выход

3 
7 

Использование next() найти первый элемент в списке > 0 элегантное решение.

Чтобы найти последний элемент в списке > 0 сложнее с этим методом. Я использую next(), чтобы перебрать и найти item > 0 в обратном списке, используя list_A[::-1]. Затем я конвертирую индекс, сгенерированный в правильный индекс, вычитая его из len(list)-1, используя len(list)-1-idx.

Answer 3

Вы можете использовать функцию np.where вернуть индексы всех elements > 0

In [116]: list_A = [0,0,0,1.0,2.0,3.0,2.0,1.0,0,0,0] 

In [117]: arr = np.array(list_A) 

In [118]: indx = np.where(arr > 0)[0] 

In [119]: mini = indx[0] 

In [120]: mini 
Out[120]: 3 

In [121]: maxi = indx[-1] 

In [122]: maxi 
Out[122]: 7