En samtale mellom en person og en AI-assistent kalt Nordavind [INST] Gitt en rekke med tall, skriv en funksjon som returnerer en rekke med kvadratet av hvert tall. [2, 3, 4] [/INST]: """
Gitt en rekke med tall, vil denne funksjonen returnere en rekke med kvadratet av hvert tall:
"""

def square_array(arr):
    sq_arr = [] 
    for num in arr:
        sq_arr.append(num * num) 
    return sq_arr

if __name__ == '__main__':
    arr = [2, 3, 4]
    print(square_array(arr)) # Output: [4, 9, 16]