Python: Make an iterator that drops elements from the iterable as long as a condition is true - w3resource
w3resource

Python: Make an iterator that drops elements from the iterable as soon as a condition is true

Python Itertools: Exercise-7 with Solution

Write a Python program to make an iterator that drops elements from the iterable as long as the elements are negative; afterwards, returns every element.

Sample Solution:

Python Code:

import itertools as it
def drop_while(nums):
    return it.takewhile(lambda x : x < 0, nums)
nums = [-1,-2,-3,4,-10,2,0,5,12]
print("Original list: ",nums)
result = drop_while(nums)
print("Drop elements from the said list as long as the elements are negative\n",list(result))
#Alternate solution
def negative_num(x):
    return x < 0
def drop_while(nums):
    return it.dropwhile(negative_num, nums)
nums = [-1,-2,-3,4,-10,2,0,5,12]
print("Original list: ",nums)
result = drop_while(nums)
print("Drop elements from the said list as long as the elements are negative\n",list(result))

Sample Output:

Original list:  [-1, -2, -3, 4, -10, 2, 0, 5, 12]
Drop elements from the said list as long as the elements are negative
 [-1, -2, -3]
Original list:  [-1, -2, -3, 4, -10, 2, 0, 5, 12]
Drop elements from the said list as long as the elements are negative
 [4, -10, 2, 0, 5, 12]

Python Code Editor:


Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Python program to make an iterator that drops elements from the iterable as soon as an element is a positive number.
Next: Write a Python program to create an iterator that returns consecutive keys and groups from an iterable.

What is the difficulty level of this exercise?

Test your Python skills with w3resource's quiz


Python: Tips of the Day