आइटम की स्थिति के आधार पर दो सूचियों पर इंडेक्स खोजें

Question

आइए कहें कि मेरे पास दो सूचियां हैं। वे -5 से 5 तक पैमाने पर पुस्तकों की रेटिंग की सूचियां हैं।

मैं जानना चाहता हूं कि list1 का तत्व >= 1 और list2 का तत्व == 0 है, उदाहरण के लिए।

list1 = [3, 3, 1, 0, 3, 0, 3, 0, 0, -3, 0, 5, 3, 0, 1, 0, 0, 5, 3, 0, 0, 0, 0, 1, 0, 3, 0, 1, 0, 0, 3, 5, 3, 3, 0, 0, 0, 5, 0, 5, 0, 3, 3, 0, -3, 0, 0, 5, 1, 5, 3, 0, 3, 0, 0] 
list2 = [5, 0, 0, 0, 0, 0, 5, 0, 0, 1, 0, 5, 3, 0, 0, 0, 0, 3, 0, 0, 0, 0, 0, 3, 0, 5, 0, 0, 0, 0, 5, 5, 5, 3, 0, 0, 0, 3, 0, 0, 0, 5, 3, 0, 0, 0, 0, 5, 0, 5, 3, 0, 0, 0, 0] 

list1[1] = 3 और list2[1] = 0, मैं जहां इस पर होता है के विभिन्न अनुक्रमित ढूँढने में सक्षम होना चाहता हूँ।

क्षमा करें अगर यह भ्रमित है लेकिन मुझे वास्तव में यह नहीं पता कि यह कैसे और शब्द है।

Answer 1

>>> [i for i, v in enumerate(list1) if v>=1 and list2[i]==0] 
[1, 2, 4, 14, 18, 27, 39, 48, 52] 

Answer 2

>>>idx_list = [i for i in range(len(list1)) if list1[i] > 1 and list2[i] == 0] 

Answer 3

मैं इस अधिक पठनीय पाया।

>>> from itertools import count 
>>> [i for i,one,two in zip(count(0), list1, list2) if one >= 1 and two == 0] 
[1, 2, 4, 14, 18, 27, 39, 48, 52] 

और यहां itertools.count doc है।

Answer 4

एक और प्रकार:

>>> [i for i, (l1, l2) in enumerate(zip(list1, list2)) if l1 >= 1 and l2 == 0] 
[1, 2, 4, 14, 18, 27, 39, 48, 52] 

Answer 5

NumPy सरणी का उपयोग करना, यह तार्किक अनुक्रमण के साथ क्या-सक्षम है:

import numpy as np 
list1 = np.array([1, -1, 0, 0, 1]) 
list2 = np.array([0, 5, 0, 0, 0]) 

# Option 1, multiply the logicals together. 
inds = np.where((list1 >= 1)*(list2 == 0))[0] 

# Option 2, pure logicals. 
inds = np.where((list1 >= 1) & (list2 == 0))[0] 

अब inds[0] = 0 और inds[1] = 4।