Python: Create group of similar items of a given list

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

Write a Python program to create groups of similar items from a given list.

Sample Solution:

Python Code:

import itertools as it 

def group_similar_items(seq):
    result =  [list(el) for _, el in it.groupby(seq, lambda x: x.split('_')[0])]
    return result 

colors = ['red_1', 'red_2', 'green_1', 'green_2', 'green_3', 'orange_1', 'orange_2']
print("Original list:")
print(colors)
print("\nGroup similar items of the said list:")
print(group_similar_items(colors))

colors = ['red_1', 'green-1', 'green_2', 'green_3', 'orange-1', 'orange_2']
print("\nOriginal list:")
print(colors)
print("\nGroup similar items of the said list:")
print(group_similar_items(colors))

Sample Output:

Original list:
['red_1', 'red_2', 'green_1', 'green_2', 'green_3', 'orange_1', 'orange_2']

Group similar items of the said list:
[['red_1', 'red_2'], ['green_1', 'green_2', 'green_3'], ['orange_1', 'orange_2']]

Original list:
['red_1', 'green-1', 'green_2', 'green_3', 'orange-1', 'orange_2']

Group similar items of the said list:
[['red_1'], ['green-1'], ['green_2', 'green_3'], ['orange-1'], ['orange_2']]

Python Code Editor:

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

Previous: Write a Python program to find all lower and upper mixed case combinations of a given string.