histogram_filling_numpy.py 526 B

1234567891011121314
  1. from __future__ import print_function
  2. import numpy as np
  3. # pip install fast-histogram
  4. from fast_histogram import histogram1d
  5. import timeit
  6. x = np.random.rand(1 << 20)
  7. nrepeat = 10
  8. print(timeit.timeit("np.histogram(x, bins=100, range=(0, 1))",
  9. "from __main__ import x, np", number=nrepeat) / (nrepeat * len(x)) / 1e-9)
  10. print(timeit.timeit("histogram1d(x, bins=100, range=(0, 1))",
  11. "from __main__ import x, histogram1d", number=nrepeat) / (nrepeat * len(x)) / 1e-9)