Python - Non-None elements indices

Many times while working in data science domain we need to get the list of all the indices which are Non None, so that they can be easily be prepossessed. This is quite a popular problem and solution to it comes quite handy. Let’s discuss certain ways in which this can be done.

Method #1 : Using list comprehension + range()
In this method we just check for each index using the range function and store the index if we find that index is Not None.

# Python3 code to demonstrate 
# Non-None elements indices 
# using list comprehension + enumerate 
  
# initializing list  
test_list = [1, None, 4, None, None, 5] 
  
# printing original list 
print("The original list : " + str(test_list)) 
  
# using list comprehension + enumerate 
# Non-None elements indices  
res = [i for i in range(len(test_list)) if test_list[i] != None] 
  
# print result 
print("The Non None indices list is : " + str(res)) 

Output :

The original list : [1, None, 4, None, None, 5]
The Non None indices list is : [0, 2, 5]

Method #2 : Using list comprehension + enumerate()
The enumerate function can be used to iterate together the key and value pair and list comprehension can be used to bind all this logic in one line.

# Python3 code to demonstrate 
# Non-None elements indices  
# using list comprehension + enumerate 
  
# initializing list  
test_list = [1, None, 4, None, None, 5] 
  
# printing original list 
print("The original list : " + str(test_list)) 
  
# using list comprehension + enumerate 
# Non-None elements indices 
res = [i for i, val in enumerate(test_list) if val != None] 
  
# print result 
print("The Non None indices list is : " + str(res)) 

Output :

The original list : [1, None, 4, None, None, 5]
The Non None indices list is : [0, 2, 5]

My Personal Notes

Python – Non-None elements indices

Recommended Posts: