Python Square todos os números na lista

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