Skip to content
Related Articles

Related Articles

Python | Index of Non-Zero elements in Python list

View Discussion
Improve Article
Save Article
Like Article
  • Last Updated : 05 Sep, 2019

Sometimes, while working with python list, we can have a problem in which we need to find positions of all the integers other than 0. This can have application in day-day programming or competitive programming. Let’s discuss a shorthand by which we can perform this particular task.

Method : Using enumerate() + list comprehension
This method can be performed using combination of functionalities. In this, we use enumerate function to access index-element together and list comprehension is used for iteration and logic creation.




# Python3 code to demonstrate working of
# Index of Non-Zero elements in Python list
# using list comprehension + enumerate()
  
# initialize list
test_list = [6, 7, 0, 1, 0, 2, 0, 12]
  
# printing original list
print("The original list is : " + str(test_list))
  
# Index of Non-Zero elements in Python list
# using list comprehension + enumerate()
res = [idx for idx, val in enumerate(test_list) if val != 0]
  
# printing result
print("Indices of Non-Zero elements : " + str(res))

Output :

The original list is : [6, 7, 0, 1, 0, 2, 0, 12]
Indices of Non-Zero elements : [0, 1, 3, 5, 7]

My Personal Notes arrow_drop_up
Recommended Articles
Page :

Start Your Coding Journey Now!