import numpy as np

c = 11*43 # c = p1*p2
k = np.random.rand(10)
a1 = np.random.rand(10).astype(complex)
a2 = np.random.rand(10).astype(complex)
b1 = np.random.rand(10).astype(complex)
b2 = np.random.rand(10).astype(complex)
lr = np.random.rand(4) * 0.0001
p2 = 1

#k = np.linspace(2,c-1,100)

corr_ = np.array([1,0])

while True:
    s_prob = c
    p1 = c / k
    p2 = c/p1
    err = c - p1*p2
    err += corr_[(p1>c*p2/p1).astype(int)] + corr_[(p2>c*p1/p2).astype(int)]
    
    k += 0.001 * err
    print(np.sum(err**2))