Создание символьного рисунка - "змейки"

length = int(input())
 
print("*" * length)
print("*" * length)
print("*" * length)

print('* **' * 5)
print('* * ' * 5)
print('*** ' * 5)

import numpy
import math
 
arr = []  # массив, где будет храниться "змейка"
count = 0  # счетчик, звездочек
up = True  # показатель, где находится звездочка
length = int(input())  # длина "змейки"
 
for i in range(math.ceil(length / 3)+2):
    if length != count:
        a = [' ']*3
        if i % 2 == 0 or i == 0:
            if count + 3 <= length:
                a = ['*']*3
                count += 3
            else:
                for t in range(length-count, 0, -1):
                    if up:
                        a[-t] = '*'
                    else:
                        a[t-1] = '*'
                count += length-count
            if up:
                up = False
            else:
                up = True
        else:
            if up:
                a[-1] = '*'
            else:
                a[0] = '*'
            count += 1
        arr.append(a)
 
a = numpy.array(arr)
for r in range(3):
    print(' '.join([str(elem)for elem in a[:, r]]))