import numpy as np 
import copy

data=np.load("data.npz", allow_pickle=True)
data=dict(data)
target_index=1
CT=data['target_ncharges'][target_index]
T=data['target_CMs'][target_index]
size_database=len(data['database_labels'])

connectivity_data = np.load("connectivity_data.npz", allow_pickle=True)
connectivity_data=dict(connectivity_data)
adjT=connectivity_data['target_adj_matrices'][target_index]

I=[]
for i in range(len(CT)):
    if CT[i] == 1:
        I.append(i)

CT=np.delete(CT, I)
T=np.delete(T, I, axis=0)
T=np.delete(T,I,axis=1)
data['target_ncharges'][target_index]=CT
data['target_CMs'][target_index]=T
adjT=np.delete(adjT, I, axis=0)
adjT=np.delete(adjT,I,axis=1)
connectivity_data['target_adj_matrices'][target_index]=adjT

for i in range(size_database):
    print("    ", i)
    adjM=connectivity_data['frag_adj_matrices'][i]
    M=data['database_CMs'][i]
    CM=data['database_ncharges'][i]
    m=len(CM)
    J=[]
    for j in range(m):
        if CM[j]==1:
            J.append(j)
    CM=np.delete(CM,J)
    M=np.delete(M,J,axis=0)
    M=np.delete(M,J,axis=1)
    adjM=np.delete(adjM,J,axis=0)
    adjM=np.delete(adjM,J,axis=1)
    connectivity_data['frag_adj_matrices'][i]=adjM
    data['database_CMs'][i]=M
    data['database_ncharges'][i]=CM

np.savez("connectivity_datafilter", fg_counts_targets=connectivity_data['fg_counts_targets'], fg_counts_frags=connectivity_data['fg_counts_frags'], target_adj_matrices=connectivity_data['target_adj_matrices'], frag_adj_matrices=connectivity_data['frag_adj_matrices'])
np.savez("datafilter", target_labels=data['target_labels'], target_CMs=data['target_CMs'], target_ncharges=data['target_ncharges'], database_labels=data['database_labels'], database_CMs=data['database_CMs'], database_ncharges=data['database_ncharges'])