import gmpy2
import matplotlib.pyplot as plt

def dec2bin(x,n):
    if x >= 0 :
        return str(bin(x))[2:].zfill(n)
    else:
        return "-" + str(bin(x))[3:].zfill(n)

def digitBalance(string):
    s = 0
    for char in string:
        if int(char) > 0:
            s += 1
        else:
            s -= 1
    return s

m = 255
primes = [1,2,3]
seq = [0,0,1]
#seq = [1,1,2]
p = 0
while p < m:
    #p = gmpy2.next_prime(primes[-1])
    p = p + 1
    b = dec2bin(p,0)
    middle = b[1:]
    print(middle)
    #middle = b
    s = digitBalance(middle)
    seq.append(seq[-1]+s)
    primes.append(p)

n = [i for i in range(len(seq))]
plt.plot(n,seq)
plt.grid(True)
plt.show()