cython_condition.py 385 B

1234567891011121314
  1. import numpy as np
  2. import pyorcy
  3. #c import cython
  4. @pyorcy.cythonize #p
  5. def condition(a, b): #p
  6. #c @cython.boundscheck(False)
  7. #c def condition(double[:] a, double[:] b):
  8. #c cdef bint [:] res = np.empty(len(a), dtype=np.int32)
  9. #c for i in range(len(a)):
  10. #c res[i] = (a[i] > 5.) & (b[i] < 1e3)
  11. #c return np.asarray(res, dtype=np.bool)
  12. return (a > 5.) & (b < 1e3)