numpy_fast.py 463 B

12345678910111213141516171819
  1. def clip(x, lo, hi):
  2. return max(lo, min(hi, x))
  3. def interp(x, xp, fp):
  4. N = len(xp)
  5. def get_interp(xv):
  6. hi = 0
  7. while hi < N and xv > xp[hi]:
  8. hi += 1
  9. low = hi - 1
  10. return fp[-1] if hi == N and xv > xp[low] else (
  11. fp[0] if hi == 0 else
  12. (xv - xp[low]) * (fp[hi] - fp[low]) / (xp[hi] - xp[low]) + fp[low])
  13. return [get_interp(v) for v in x] if hasattr(x, '__iter__') else get_interp(x)
  14. def mean(x):
  15. return sum(x) / len(x)