import itertools def abc(Y,X,N,p): for dy,dx in itertools.product(range(Y),range(-X+1,X)): if dy==dx==0: continue N-=1 for y,x in itertools.product(range(Y),range(X)): if not 0<=y+dy