__author__ = 'danil.gizdatullin'

# import numpy as np
import matplotlib.pyplot as plt
import time

from k_nearest_neighbors import KNearestNeighbors


def special_print(a):
    for key in a.iterkeys():
        print("Key = %i" % key)
        print(a[key].dict)
        print("#####################")

start_time = time.time()

k_nn = KNearestNeighbors()
# special_print(k_nn.k_nearest_dict)
# special_print(k_nn.reverse_k_nearest_dict)
distr = k_nn.return_distribution_k_nearest_neighbors()
distr_reverse = k_nn.return_distribution_reverse_k_nearest_neighbors()

print("--- %s seconds ---" % (time.time() - start_time))
# plt.hist(distr)
plt.hist(distr_reverse, bins=100)
plt.show()