Python: Make an iterator that returns consecutive keys and groups from the iterable

Last update on December 21 2024 07:31:08 (UTC/GMT +8 hours)

Write a Python program to create an iterator that returns consecutive keys and groups from an iterable.

Sample Solution:

Python Code:

import itertools as it
print("Iterate over characters of a string and display\nconsecutive keys and groups from the iterable:")
str1 = 'AAAAJJJJHHHHNWWWEERRRSSSOOIIU'
data_groupby = it.groupby(str1)
for key, group in data_groupby:
    print('Key:', key)
    print('Group:', list(group))    
print("\nIterate over elements of a list and display\nconsecutive keys and groups from the iterable:")
str1 = 'AAAAJJJJHHHHNWWWEERRRSSSOOIIU'    
str1 = [1,2,2,3,4,4,5,5,5,6,6,7,7,7,8]
data_groupby = it.groupby(str1)
for key, group in data_groupby:
    print('Key:', key)
    print('Group:', list(group))

Sample Output:

Iterate over characters of a string and display
consecutive keys and groups from the iterable:
Key: A
Group: ['A', 'A', 'A', 'A']
Key: J
Group: ['J', 'J', 'J', 'J']
Key: H
Group: ['H', 'H', 'H', 'H']
Key: N
Group: ['N']
Key: W
Group: ['W', 'W', 'W']
Key: E
Group: ['E', 'E']
Key: R
Group: ['R', 'R', 'R']
Key: S
Group: ['S', 'S', 'S']
Key: O
Group: ['O', 'O']
Key: I
Group: ['I', 'I']
Key: U
Group: ['U']

Iterate over elements of a list and display
consecutive keys and groups from the iterable:
Key: 1
Group: [1]
Key: 2
Group: [2, 2]
Key: 3
Group: [3]
Key: 4
Group: [4, 4]
Key: 5
Group: [5, 5, 5]
Key: 6
Group: [6, 6]
Key: 7
Group: [7, 7, 7]
Key: 8
Group: [8]

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 long as the elements are negative; afterwards, returns every element.
Next: Write a Python program to split an iterable and generate iterables specified number of times.