python list comprehension square code example

Example: list comprehension with square numbers python

def square(a):
    squares = []
    for i in a:
        squares.append(i**2)
    return squares