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
- New Content published on w3resource:
- Scala Programming Exercises, Practice, Solution
- Python Itertools exercises
- Python Numpy exercises
- Python GeoPy Package exercises
- Python Pandas exercises
- Python nltk exercises
- Python BeautifulSoup exercises
- Form Template
- Composer - PHP Package Manager
- PHPUnit - PHP Testing
- Laravel - PHP Framework
- Angular - JavaScript Framework
- React - JavaScript Library
- Vue - JavaScript Framework
- Jest - JavaScript Testing Framework