Python tìm ra chỉ số của một danh sách được đưa ra một điều kiện

Question

tôi có một danh sách, chúng ta hãy nói:

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

Tôi muốn tìm ra tối thiểu và tối đa chỉ số của danh sách này, nơi list_A> 0.

tức là trong ví dụ trên, nó sẽ là 3 và 7.

Đối với các danh sách khác, làm tăng đơn điệu, tôi đã được sử dụng np.searchsorted, như np.searchsorted(list,[0.5,1.0]) để tìm các chỉ số trong danh sách tương ứng là between 0.5 and 1.0.

Nhưng trường hợp này hoàn toàn khác và np.searchsorted không hoạt động ở đây hoặc có thể theo cách mà tôi không biết!

Answer 1

Lọc danh sách nén với indixes của nó và lấy min và max:

>>> 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) 

Nếu bạn chỉ cần chỉ mục, giải nén giá trị trả về:

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

Answer 2

Một thay thế sẽ được sử dụng 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)) 

Đầu ra

3 
7 

Sử dụng next() để tìm mục đầu tiên trong danh sách > 0 là một giải pháp thanh lịch.

Để tìm số mục cuối cùng trong danh sách > 0 thì phức tạp hơn với phương pháp này. Tôi sử dụng next() để lặp lại và tìm số đầu tiên mục > 0 trong danh sách được đảo ngược bằng cách sử dụng list_A[::-1]. Sau đó tôi chuyển đổi chỉ mục được tạo thành chỉ mục chính xác bằng cách trừ chỉ mục từ len(list)-1, sử dụng len(list)-1-idx.

Answer 3

Bạn có thể sử dụng np.where hàm trả về chỉ số của tất cả các 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