w3resource

Python: Insert an element in a given list after every nth position

Python List: Exercise - 170 with Solution

Write a Python program to insert an element in a given list after every nth position.

Visual Presentation:

Python List: Insert an element in a given list after every nth position.

Sample Solution:

Python Code:

# Define a function called 'insert_elemnt_nth' that inserts an element 'ele' into a list 'lst' after every 'n' elements.
def insert_elemnt_nth(lst, ele, n):
    # Create an empty list 'result' to store the modified list.
    result = []
    # Iterate over the list with a step of 'n' using a range.
    for st_idx in range(0, len(lst), n):
        # Extend 'result' with the elements from the current slice of 'lst' and add 'ele'.
        result.extend(lst[st_idx:st_idx+n])
        result.append(ele)
    # Remove the last occurrence of 'ele' to correct any extra insertions.
    result.pop()
    return result

# Create a list 'nums' containing integer values.
nums = [1, 2, 3, 4, 5, 6, 7, 8, 9, 0]

# Print a message indicating the original list.
print("Original list:")
print(nums)

# Set the element 'i_ele' to 'a' and the insertion position 'i_ele_pos' to 2.
i_ele = 'a'
i_ele_pos = 2
print("\nInsert", i_ele, "in the said list after", i_ele_pos, "nd element:")

# Call the 'insert_elemnt_nth' function to insert 'i_ele' into 'nums' after every 2nd element and print the result.
print(insert_elemnt_nth(nums, i_ele, i_ele_pos))

# Set the element 'i_ele' to 'b' and the insertion position 'i_ele_pos' to 4.
i_ele = 'b'
i_ele_pos = 4
print("\nInsert", i_ele, "in the said list after", i_ele_pos, "th element:")

# Call the 'insert_elemnt_nth' function to insert 'i_ele' into 'nums' after every 4th element and print the result.
print(insert_elemnt_nth(nums, i_ele, i_ele_pos))  

Sample Output:

Original list:
[1, 2, 3, 4, 5, 6, 7, 8, 9, 0]

Insert a in the said list after 2 nd element:
[1, 2, 'a', 3, 4, 'a', 5, 6, 'a', 7, 8, 'a', 9, 0]

Insert b in the said list after 4 th element:
[1, 2, 3, 4, 'b', 5, 6, 7, 8, 'b', 9, 0]

Flowchart:

Flowchart: Insert an element in a given list after every nth position.

Python Code Editor:

Previous: Write a Python program to convert a given list of strings and characters to a single list of characters.
Next: Write a Python program to concatenate element-wise three given lists.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Become a Patron!

Follow us on Facebook and Twitter for latest update.

It will be nice if you may share this link in any developer community or anywhere else, from where other developers may find this content. Thanks.

https://www.w3resource.com/python-exercises/list/python-data-type-list-exercise-170.php