Python Find Closest Element In Array