Python | Index of Non-Zero elements in Python list
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 listtest_list = [6, 7, 0, 1, 0, 2, 0, 12] # printing original listprint("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 resultprint("Indices of Non-Zero elements : " + str(res)) |
The original list is : [6, 7, 0, 1, 0, 2, 0, 12] Indices of Non-Zero elements : [0, 1, 3, 5, 7]
Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.
To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course. And to begin with your Machine Learning Journey, join the Machine Learning – Basic Level Course
