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. 

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

Time complexity: O(n), where n is the length of the input list.
Auxiliary space:  O(n).

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. 

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

Time complexity: O(n), where n is the length of the input list test_list. This is because the code iterates through each element of the list once.
Auxiliary space: O(k), where k is the number of non-None elements in the input list. 

Method #3: Using the built-in functions zip, range, filter, and lambda

Using the built-in functions zip, range, filter, and lambda, we can create a list of non-None indices by first zipping the original list with the range of its length, then filtering out the elements where the original list value is None, and finally extracting the indices using a lambda function.

Here’s an example of how this can be done:

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

Time complexity: O(n)
Auxiliary Space: O(n)

Method 4: Using numpy

• Import the numpy library: import numpy as np
• Convert the list to a numpy array: arr = np.array(test_list). This creates a new numpy array from the original list.
• Get the indices where the array elements are not None: non_none_indices = np.where(arr != None)[0]. The where() function returns a tuple containing the indices where the specified condition is true. Since we only need the first element of the tuple (which contains the indices), we append [0] to the end of the function call.
• Convert the numpy array to a list: res = non_none_indices.tolist(). This creates a new list from the numpy array containing the non-None indices.
• Print the result: print(“The Non None indices list is : ” + str(res)). This displays the list of non-None indices in the output.

Output:

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

Time complexity: O(n)
Auxiliary Space: O(n)

Method 5 : Using a for loop

step by step approach :

• Create a list of integers and None values named test_list. The list contains 6 elements: [1, None, 4, None, None, 5].
• Create an empty list named non_none_indices. This list will store the indices of non-None elements from the test_list.
• Loop through the range of indices of test_list using a for loop. The range of indices is obtained using the range() function, which takes the length of test_list as its argument. This loop will iterate through all the elements of test_list.
• Check if the element at the current index in test_list is not None using an if statement. If it’s not None, then append the index of that element to the non_none_indices list.
• After iterating through all the elements of test_list, print the non_none_indices list using the print() function. The list will contain the indices of non-None elements from the test_list.

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

Time Complexity: O(n) where n is the length of the list.
Auxiliary Space: O(k) where k is the number of non-None elements in the list.

The time complexity is O(n) because we are iterating over the entire list once. The auxiliary space is O(k) because we are storing the indices of non-None elements in a separate list, which can have a maximum size of k.

manjeet_04

Follow

Improve

Article Tags :

• Python
• Python Programs
• Python list-programs

Practice Tags :

• python