Python | Group List on K character
Last Updated :
09 Apr, 2023
Sometimes, we may face an issue in which we require to split a list to list of list on the K character sent as deliminator. This kind of problem can be used to send messages or can be used in cases where it is desired to have list of list of native list. Let’s discuss certain ways in which this can be done.
Method #1: Using index() and list slicing The list slicing can be used to get the sublists from the native list and index function can be used to check for the K character which can potentially act as a separator. The drawback of this is that it only works for a single split i.e can only divide a list to 2 sublist.
Python3
test_list = [ 'Geeks' , 'for' , 'M' , 'Geeks' , 1 , 2 ]
print ( "The original list : " + str (test_list))
K = 'M'
temp_idx = test_list.index(K)
res = [test_list[: temp_idx], test_list[temp_idx + 1 : ]]
print ( "The list of sublist after separation : " + str (res))
|
Output :
The original list : ['Geeks', 'for', 'M', 'Geeks', 1, 2]
The list of sublist after separation : [['Geeks', 'for'], ['Geeks', 1, 2]]
Time Complexity: O(n) where n is the number of elements in the list “test_list”.
Auxiliary Space: O(n) where n is the number of elements in the list “test_list”.
Method #2: Using loops The problem of the above proposed method can be solved using general loops and using brute force, in this we just look for K character and make a new list after that.
Python3
test_list = [ 'Geeks' , 'M' , 'for' , 'M' , 4 , 5 , 'M' , 'Geeks' , 'CS' , 'M' , 'Portal' ]
print ( "The original list : " + str (test_list))
K = 'M'
temp = []
res = []
for ele in test_list:
if ele = = K:
res.append(temp)
temp = []
else :
temp.append(ele)
res.append(temp)
print ( "The list of sublist after separation : " + str (res))
|
Output :
The original list : ['Geeks', 'for', 'M', 'Geeks', 1, 2]
The list of sublist after separation : [['Geeks', 'for'], ['Geeks', 1, 2]]
Time Complexity: O(n*n), where n is the number of elements in the list “test_list”.
Auxiliary Space: O(n), where n is the number of elements in the list “test_list”.
Method #3: Using itertools.groupby()
Python3
import itertools
test_list = [ 'Geeks' , 'for' , 'M' , 'Geeks' , 1 , 2 ]
print ( "The original list : " + str (test_list))
K = 'M'
res = [ list (g) for k, g in itertools.groupby(test_list, lambda x: x = = K) if not k]
print ( "The list of sublist after separation : " + str (res))
|
Output
The original list : ['Geeks', 'for', 'M', 'Geeks', 1, 2]
The list of sublist after separation : [['Geeks', 'for'], ['Geeks', 1, 2]]
Time Complexity: O(n)
Auxiliary Space: O(n)
Share your thoughts in the comments
Please Login to comment...