Find Closest Element In Array Python Initialize 2d