InsertAtPosition.py

def insert_at_pos(head, data, position):
    # edge case: check if pos is 0
    new_node = Node(data)
    if position is 0:
        new_node.next = head
        head = new_node
        return head

    pos = 0
    new_node = Node(data)
    curr_node = head
    # iterating till the position is reached
    while curr_node.next and pos < position - 1:
        pos += 1
        curr_node = curr_node.next
    # making new node's next as current node's next
    new_node.next = curr_node.next
    # making current node's next point to new node
    curr_node.next = new_node
    return head