summaryrefslogtreecommitdiffstats
path: root/src/boost/libs/histogram/benchmark/histogram_filling_numpy.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/boost/libs/histogram/benchmark/histogram_filling_numpy.py')
-rw-r--r--src/boost/libs/histogram/benchmark/histogram_filling_numpy.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/src/boost/libs/histogram/benchmark/histogram_filling_numpy.py b/src/boost/libs/histogram/benchmark/histogram_filling_numpy.py
new file mode 100644
index 00000000..ce0fa2b7
--- /dev/null
+++ b/src/boost/libs/histogram/benchmark/histogram_filling_numpy.py
@@ -0,0 +1,14 @@
+from __future__ import print_function
+import numpy as np
+# pip install fast-histogram
+from fast_histogram import histogram1d
+import timeit
+
+x = np.random.rand(1 << 20)
+nrepeat = 10
+
+print(timeit.timeit("np.histogram(x, bins=100, range=(0, 1))",
+ "from __main__ import x, np", number=nrepeat) / (nrepeat * len(x)) / 1e-9)
+
+print(timeit.timeit("histogram1d(x, bins=100, range=(0, 1))",
+ "from __main__ import x, histogram1d", number=nrepeat) / (nrepeat * len(x)) / 1e-9) \ No newline at end of file