26 lines
370 B
Python
26 lines
370 B
Python
import numpy as np
|
|
|
|
a = np.array([1,2,3,4])
|
|
|
|
print(a)
|
|
|
|
import time
|
|
a = np.random.rand(1000000)
|
|
b = np.random.rand(1000000)
|
|
|
|
tic = time.time()
|
|
c = np.dot(a,b)
|
|
toc = time.time()
|
|
|
|
print("Vectorized version:" + str(1000*(toc-tic)) + "ms")
|
|
|
|
c = 0
|
|
tic = time.time()
|
|
for i in range(1000000):
|
|
c += a[i]*b[i]
|
|
toc = time.time()
|
|
|
|
print("For loop:" + str(1000*(toc-tic)) + "ms")
|
|
|
|
|