nsp.py 401 B

123456789101112
  1. import numpy as np
  2. def fix_timestamps(tstmp, offsets=None):
  3. tstmp = tstmp.astype(np.int64)
  4. rollover, = np.where(np.diff(tstmp) < 0)
  5. if offsets is None:
  6. offsets = tstmp[rollover]
  7. for j, i in enumerate(rollover):
  8. tstmp[(i + 1):] = tstmp[(i + 1):] + offsets[j]
  9. # offsets[j:] += tstmp[i]
  10. cum_offsets = np.cumsum(offsets)
  11. return tstmp, offsets, cum_offsets