Python | Adding value to sublists

# Python3 code to demonstrate
# appending single value
# using list comprehension

# initializing list of lists
test_list = [[1, 3], [3, 4], [6, 5], [4, 5]]

# printing original list
print("The original list : " + str(test_list))

# declaring element to be inserted
K = "GFG"

# using list comprehension
# appending single value
res = [[i, j, K] for i, j in test_list]

# printing result
print("The list after adding element :  " + str(res))

# Python3 code to demonstrate
# appending single value
# using list comprehension + "+" operator

# initializing list of lists
test_list = [[1, 3], [3, 4], [6, 5], [4, 5]]

# printing original list
print("The original list : " + str(test_list))

# declaring element to be inserted
K = "GFG"

# using list comprehension + "+" operator
# appending single value
res = [sub + [K] for sub in test_list]

# printing result
print("The list after adding element :  " + str(res))

# Python3 code to demonstrate
# appending single value

# initializing list of lists
test_list = [[1, 3], [3, 4], [6, 5], [4, 5]]

# printing original list
print("The original list : " + str(test_list))

# declaring element to be inserted
K = "GFG"
res = []
for i in test_list:
    i.append(K)
    res.append(i)


# printing result
print("The list after adding element : " + str(res))

# Python3 code to demonstrate
# appending single value
 
# initializing list of lists
test_list = [[1, 3], [3, 4], [6, 5], [4, 5]]
# printing original list
print("The original list : " + str(test_list))
# Declaring element to be inserted
K = "GFG"

# Using map() and lambda function to add element to all sublists
res = list(map(lambda x: x + [K], test_list))
print("The list after adding element :  " + str(res))

#This code is contributed by Edula Vinay Kumar Reddy

import itertools  # Import the itertools module
test_list = [[1, 3], [3, 4], [6, 5], [4, 5]]  # Initialize the list of lists
K = "GFG"  # Declare the element to be inserted
# printing original list
print("The original list : " + str(test_list))
res = [list(itertools.chain(i, [K])) for i in test_list]
# Use list comprehension and itertools.chain to add the element K to each sublist in the list of lists
# itertools.chain takes multiple iterables as arguments and returns a single, flatten iterable that contains elements from all the input iterables
# Here, i is each sublist in test_list, and [K] is a list with a single element K
# The result of itertools.chain(i, [K]) is a flatten list that contains all elements in i followed by K
# The list comprehension [list(itertools.chain(i, [K])) for i in test_list] applies this operation to each sublist in test_list and collects the results into a new list of lists
print("The list after adding element :  " + str(res))  # Print the result
# This code is contributed by Jyothi pinjala.

import numpy as np

# initializing list of lists
test_list = [[1, 3], [3, 4], [6, 5], [4, 5]]

# printing original list
print("The original list : " + str(test_list))

# declaring element to be inserted
K = "GFG"

# using numpy
# appending single value
test_np = np.array(test_list)
res_np = np.append(test_np, np.full((len(test_list), 1), K), axis=1).reshape(len(test_list), -1).tolist()

# printing result
print("The list after adding element :  " + str(res_np))

test_list = [[1, 3], [3, 4], [6, 5], [4, 5]]
K = "GFG"

for sublist in test_list:
    sublist.extend([K])

print("The list after adding element : " + str(test_list))