from math import sqrt def point(x, y): distance = 300 if x<500 and sqrt((x-300)**2+y**2)> distance: return 2 elif x > 500 and sqrt((x - 300)**2+(y + 1000)**2)> distance: return 2 else: return 3