function - matrix/array

def create_matrix(square_size):
    array = []
    for i in range(square_size):
        array.append([])
        for j in range(square_size):
            array[i].append(0)
    return array