Python Find Closest Value In Numpy Which Function