квадрат все элементы в списке Python

def square(list):
    return [i ** 2 for i in list]
Ugliest Unicorn