1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
#!/usr/bin/env python

import numpy as n

print "Exercise (a)"
a = n.array ( [[1, 2, 3], [2, 6, 10],[1, 3, 5]] )
A = n.matrix(a)
# alternatively: A = n.asmatrix(a)
print A

print "Exercise (b)"
B = n.mat ( '1 2 3-1.5j; 4 5 6; 1.2 2.3 3.4' )
print B

print "Exercise (c)"
print "A + B =", A+B
print "a + b =", n.asarray(A)+n.asarray(B)
print "A * B =", A*B
print "a * b =", n.asarray(A)*n.asarray(B)

print "Exercise (d)"
A = n.mat( "1 0 -1; 2 5 3; -4 0 1")
v = n.array((4,-5,13))
x = n.linalg.solve(A,v)
print "A * x = v -> x =", x