example3.py

import numpy as np
import time
a = (np.random.rand(1,1000000)) #create a random (1,10) array
result1 = np.zeros((1,1000000),dtype=np.int) #create an array of zeros to hold the result
#print ("a: \n" + str(a))
tic = time.time()
for i in range(1000000):
    if a[0,i]>0.5:
        result1[0,i] = 1
    else:
        result1[0,i] = 0
#print ("result: \n" +str(result))
toc = time.time()
print ("time passed for result1: " + str(toc-tic) + "ms")

tic = time.time()
result2 = (np.where(a>0.5,1,0)) 
#print ("result2: \n" +str(result2))
toc = time.time()
print ("time passed for result2: " + str(toc-tic) + "ms")

assert((result1 == result2).all)