In this article, we will be going to understand how to get all the indices of an element in a list in Python and we will also cover a few of the best approaches to do so.
Example
Input: my_list = [1, 2, 3, 1, 5, 4]
item = 1
Output: 0,3
Explanation: The output contains all the index of element 1 in the my_list.
Find List Index of All Occurrences of an Element
Below are the topics that we will cover in this article:
- Using a Loop
- Using List Comprehension
- Using Enumerate() Function
- Using itertools module
- Using Numpy library
- Using the Python Dictionary
- Using the Panda library
Find the Index of All Occurrences of an Item Using a Loop
This method to get the indices of all occurrences of an element in a list. Here we use a for-loop to iterate through each element in the original list.
Python3
# initialize a listmy_list = [1, 2, 3, 1, 5, 4]# find length of the listlist_size = len(my_list)# declare for loopfor itr in range(list_size): # check the condition if(my_list[itr] == 1): # print the indices print(itr) |
0 3
Time Complexity: O(n)
Space Complexity: O(1)
Get Index of All Occurrences of Element Using List Comprehension
This is a simple method to get the indices of all occurrences of an element in a list using list comprehension. Here we use a list comprehension to iterate through each element in the original list.
Python3
# initialize a listmy_list = [1, 2, 3, 1, 5, 4]item = 1indices = [i for i in range(len(my_list)) if my_list[i] == item]# print the indicesprint(indices) |
[0, 3]
Time Complexity: O(n)
Auxiliary Space: O(1)
Find Index of All Occurrences of an Item Using Enumerate() Function
Instead of using for-loop we can use enumerate function. This function adds a counter to an iterable and returns the enumerated object. For this, we will create a list and then we will create enumerate function to get indices of all occurrences of an element in a list
Python3
# initialize a listmy_list = [1, 2, 3, 1, 5, 4] indices = [ind for ind, ele in enumerate(my_list) if ele == 1]# print the indicesprint(indices) |
Time Complexity: O(n)
Space Complexity: O(n)
Find index of All Occurrences of an Item Using Itertools Module
An itertools module is a memory-efficient tool that is useful by itself or in combination, so for this, we will use count() methods from this module which will return an iterator of evenly spaced values from the start value. For this, we will create a list, and then using comprehension with zip() we will get indices of all occurrences of an element in a list.
Python3
# import count method from itertoolsfrom itertools import count # initialize a listmy_list = [1, 2, 3, 1, 5, 4]indices = [ind for ind, ele in zip(count(), my_list) if ele == 1]# print the indicesprint(indices) |
[0, 3]
Time Complexity: O(n)
Space Complexity: O(n)
Get the Indices of All Occurrences of an Element Using Numpy Library
NumPy is a general-purpose array-processing package, it provides convenient ways to use arrays in Python. For this, we will create an array using numpy.array() and then get all the indices of elements in an input array when the condition is satisfied using numpy.where() methods.
Python3
# import numpy moduleimport numpy # initialize a arraymy_list = numpy.array([1, 2, 3, 1, 5, 4]) indices = numpy.where(my_list == 1)[0]# display resultprint(indices) |
Output :
[0, 3]
Time Complexity: O(n)
Space Complexity: O(n)
Get the indices of all occurrences of an element using the Python Dictionary
A dictionary called `indices` and a list named `my_list`. It then iterates through the elements in `my_list`, storing their indices in the dictionary `indices`. For each element encountered, if it’s already a key in the dictionary, the current index is appended to its list of indices; otherwise, a new key is created with the element and its index. Afterward, the code prints the indices of the value 1 in the list. As a result, the indices of the value 1 in `my_list` (0 and 3) are displayed when the code is executed.
Python3
indices = {}# initialize a listmy_list = [1, 2, 3, 1, 5, 4]for i in range(len(my_list)): if my_list[i] in indices.keys(): indices[my_list[i]].append(i) else: indices[my_list[i]] = [i]# print the indicesprint(indices[1]) |
[0, 3]
Get the Indices of All Occurrences of an Element Using the Panda Library
Import the pandas library then Initialize the list “my_list” and the item whose indices are to be found and create a pandas dataframe “df” with a column “col” containing the elements of the list. Filter the dataframe rows based on the value of the “col” column matching the given item.Get the indices of the filtered rows and then convert the pandas index object to a list and print the list of indices
Python3
import pandas as pdmy_list = [1, 2, 3, 1, 5, 4]item = 1df = pd.DataFrame({'col': my_list})indices = df.index[df['col'] == item].tolist()print(indices) |
[0, 3]
Time complexity: O(n), where n is the number of elements in the input list.This is because creating the pandas data frame takes O(n) time, and filtering the data frame takes O(n) time in the worst case.
Auxiliary Space: O(n), where n is the number of elements in the input list. This is because creating the pandas data frame requires O(n) space to store the elements of the list.
