import numpy as np from scipy.spatial.distance import cdist def nearest_neighbor(samples, targets, samples_to_classify, metric='euclidean'): return targets[np.argmin(cdist(samples_to_classify, samples, metric=metric), axis=1)]