def foo(): if n := randint(0, 3): return n ** 2 return 1337 [(x, y, x/y) for x in input_data if (y := f(x)) > 0]
def foo(): if randint(0, 3) as n: return n ** 2 return 1337