Python – Convert list into list of lists



Python | Convert list into list of lists

Given a list of strings, write a Python program to convert each element of the given list into a sublist. Thus, converting the whole list into a list of lists.

Examples:

 

Input : ['alice', 'bob', 'cara']
Output : [['alice'], ['bob'], ['cara']]

Input : [101, 202, 303, 404, 505] 
Output : [[101], [202], [303], [404], [505]]

Approach #1 : Naive Approach

Use another list ‘res’ and a for a loop. Using split() method of Python we extract each element from the list in the form of the list itself and append it to ‘res’. Finally, return ‘res’. One drawback of this method is that it does not work with integer list as ‘int’ object has no attribute ‘split’.

# Python3 program to convert 
# list into a list of lists
 
def extractDigits(lst):
    res = []
    for el in lst:
        sub = el.split(', ')
        res.append(sub)
     
    return(res)
                 
# Driver code
lst = ['alice', 'bob', 'cara']
print(extractDigits(lst))

Output:

[['alice'], ['bob'], ['cara']]

Approach #2 : List comprehension

List comprehension is an efficient approach as it doesn’t make use of extra space. For each element ‘el’ in list, it simply appends [el] to the output list.

# Python3 program to convert 
# list into a list of lists
 
def extractDigits(lst):
    return [[el] for el in lst]
                 
# Driver code
lst = ['alice', 'bob', 'cara']
print(extractDigits(lst))

Output:

[['alice'], ['bob'], ['cara']]

Approach #3 : Python map()

The given code maps the function el:[el] for each item of the given iterable ‘lst’. Hence outputs each element as a list itself.

# Python3 program to convert 
# list into a list of lists
 
def extractDigits(lst):
    return list(map(lambda el:[el], lst))
     
             
# Driver code
lst = ['alice', 'bob', 'cara']
print(extractDigits(lst))

Output:

[['alice'], ['bob'], ['cara']]

Last Updated on March 1, 2022 by admin

Leave a Reply

Your email address will not be published. Required fields are marked *

Recommended Blogs