KolakoskiSequenceを生成するプログラム

kolakoski_sequence.py

# coding: UTF-8

def kolakoski_transformation(seq):
    l = []
    temp = seq[0]
    count = 0
    for n in seq:
        if(n == temp):
            count+=1
        else:
            l.append(count)
            temp = n
            count =1
    l.append(count)
    return l

def kolakoski_sequence(integers,length):
    l = []
    index = 0
    while len(l)<length:
        for n in integers:
            if(len(l)<=index):
                l.extend([n]*n)
                index+=1
            else:
                l.extend([n]*l[index])
                index+=1
    return l[:length]

seq =kolakoski_sequence([1,2],100)
print(seq)
print(kolakoski_transformation(seq))