Python Grouping similar substrings in list

PythonServer Side ProgrammingProgramming

In this tutorial, we are going to write a program that groups similar substrings from a list. Let's see an example to understand it more clearly.

Input

strings = ['tutorials-python', 'tutorials-c', 'tutorials-java', 'tutorials-javascript', 'python-1', 'python-2',
'javascript-1']

Output

[['tutorials-python', 'tutorials-c', 'tutorials-java', 'tutorials-javascript'], ['python-1', 'python-2'],
['javascript-1']]

We are going to use the groupby method from itertools module to solve the problem. The groupby method will group all the similar string into an iter object. For the given list we have split the string with - and pass the first part of the string to the groupby method.

Let's see the steps involved in solving this problem.

  • Initialize the list of strings.
  • Import the itertools module.
  • Initialize an empty list.
  • Now, pass the strings and a lambda function to the itertools.groupby method.
  • The lambda function should return first part of the string after splitting it with −
  • The groupby method will return a list of the tuples with the element and its group.
  • In every iteration, convert the group of similar elements into a list.
  • Append the list to the empty list.
  • Print the result.

Example

 Live Demo

# importing the itertools module
import itertools
# initializing the strings
strings = ['tutorials-python', 'tutorials-c', 'tutorials-java', 'tutorials-javascript', 'python-1', 'python-2', 'javascript-1']
# empty list
result = []
# iterator
# lambda function will return first part from the string
iterator = itertools.groupby(strings, lambda string: string.split('-')[0])
# iterating over the result
# element and its group
for element, group in iterator:
   # appending the group by converting it into a list
   result.append(list(group))
# printing the result
print(result)

Output

If you run the above code, then you will get the following result.

[['tutorials-python', 'tutorials-c', 'tutorials-java', 'tutorials-javascript'],
ython-1', 'python-2'], ['javascript-1']]

Conclusion

If you have any doubts in the tutorial, mention them in the comment section.

raja
Published on 07-Jul-2020 08:38:51
Advertisements