random_profile.py

import timeit

number = 10000
numpy_time = timeit.timeit("[np.random.rand() for _ in range(int(1e3))]", "import numpy as np", number=number)
random_time = timeit.timeit("[random.random() for _ in range(int(1e3))]", "import random", number=number)
numpy_batch_time = timeit.timeit("np.random.rand(int(1e3))", "import numpy as np", number=number)

print("Timings")
print("=======")
print(f"Numpy Single: {numpy_time:.3f}")
print(f"Random:       {random_time:.3f}")
print(f"Numpy Batch:  {numpy_batch_time:.3f}")
print("=======")

# =======
# Numpy Single: 3.245
# Random:       1.003
# Numpy Batch:  0.085
# =======