b.py

import gmpy2
import numpy as np
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 = 10000
primes = [1,2,3]
seq = [0,0,1]
#seq = [1,1,2]
p = 3
while p < m:
    p = gmpy2.next_prime(primes[-1])
    #p += 1
    b = dec2bin(p,0)
    middle = b[1:-1]
    #left = b[:-1]
    #right = b[1:]
    #s = digitBalance(left)
    #s = digitBalance(right)
    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()