Мне сказали

Напишите функцию square (a), которая принимает массив чисел a и возвращает массив, содержащий каждое из значений квадрата.

Сначала я имел

def square(a):
    for i in a: print i**2

Но это не работает, так как я печатаю и не возвращаюсь, как меня просили. Так я попробовал

    def square(a):
        for i in a: return i**2

Но это только квадраты последнего числа моего массива. Как я могу получить это, чтобы уместить весь список?