from __future__ import division
from scipy import *
import time

array_size = [3,5]

x = random.randint(0, 10, size=array_size)
start_time = time.time()
y = sin(x) ** 2 + cos(x)**2
stop_time = time.time()
print y
array_seconds = stop_time-start_time
print array_seconds, 'seconds'

print

# for loop
x = random.randint(0, 10, size=array_size)
y = zeros(shape(x)) # create a zero array of same size as x
start_time = time.time()
for i in range(shape(x)[0]):
    for j in range(shape(x)[1]):
        y[i,j] = sin(x[i,j]) ** 2 + cos(x[i,j])**2
stop_time = time.time()
print x
print y
for_loop_seconds = stop_time-start_time
print for_loop_seconds, 'seconds'

print
print for_loop_seconds / array_seconds


