from IPython.display import HTML

HTML('<img src="http://teaching.software-carpentry.org/wp-content/uploads/2012/11/array-math-cmap.png" height="500px">')

import numpy as np

a = np.arange(5)
b = np.arange(5, 10)
print a
print b

a + 6

new_a = []
for i in xrange(a.size):
    new_a.append(a[i] + 6)
np.array(new_a)

a * b

a = np.arange(5)
b = np.arange(6)
a * b

a = np.ones((3, 2)) # 3 x 2
b = np.arange(4, 6) # 1 x 2
a * b

b = np.array([[4], [5], [6]]) # 3 x 1
a * b

a = np.ones((4, 2)) # 4 rows, 2 columns
b = np.array([[4, 5], [6, 7]]) # 2 rows, 2 columns
a * b

a = np.ones((4, 3, 2))
b = np.array([[4, 5], [6, 7], [8, 9]]) # 3 x 2
a * b