こういう技を使うと論理や符号の勘違いでバグ起こしやすいから、素直にif文使ったほうがいい
Pythonでどうしても高速化したいんなら、CupyでGPU演算するとか、Cythonでバイナリ化するとか考えるべき