import numpy as np

N=4
M=np.matrix([[0,1,0,0],[1,0,1,0],[0,1,0,1],[0,0,1,1]])
V=np.linalg.eig(M)
print(V)

qn=[1 for i in range(N)]
for i in range(N):
    print('E=',V[0][i])
    for j in range(N):
        print('C',j+1,' = ',V[1][j,i])
        if i>=2:
            qn[j]-= 2*pow(V[1][j,i],2)
print('charges: ')
for i in range(N):
    print('atome',i+1,' = ',qn[i])