这篇文章主要介绍了python实现的矩阵类,结合完整实例形式分析了python矩阵的定义、计算、转换等相关操作技巧,需要的朋友可以参考下
本文实例讲述了python实现的矩阵类。分享给大家供大家参考，具体如下：
科学计算离不开矩阵的运算。当然，python已经有非常好的现成的库：numpy(numpy的简单安装与使用
我写这个矩阵类，并不是打算重新造一个轮子，只是作为一个练习，记录在此。
注：这个类的函数还没全部实现，慢慢在完善吧。
全部代码：
import copy
class matrix:
'''矩阵类'''
def __init__(self, row, column, fill=0.0):
self.shape = (row, column)
self.row = row
self.column = column
self._matrix = [[fill]*column for i in range(row)]
# 返回元素m(i, j)的值: m[i, j]
def __getitem__(self, index):
if isinstance(index, int):
return self._matrix[index-1]
elif isinstance(index, tuple):
return self._matrix[index[0]-1][index[1]-1]
# 设置元素m(i,j)的值为s: m[i, j] = s
def __setitem__(self, index, value):
if isinstance(index, int):
self._matrix[index-1] = copy.deepcopy(value)
elif isinstance(index, tuple):
self._matrix[index[0]-1][index[1]-1] = value
def __eq__(self, n):
'''相等'''
# a == b
assert isinstance(n, matrix), "类型不匹配，不能比较"
return n.shape == self.shape # 比较维度，可以修改为别的
def __add__(self, n):
'''加法'''
# a + b
assert n.shape == self.shape, "维度不匹配，不能相加"
m = matrix(self.row, self.column)
for r in range(self.row):
for c in range(self.column):
m[r, c] = self[r, c] + n[r, c]
return m
def __sub__(self, n):
'''减法'''
# a - b
assert n.shape == self.shape, "维度不匹配，不能相减"
m = matrix(self.row, self.column)
for r in range(self.row):
for c in range(self.column):
m[r, c] = self[r, c] - n[r, c]
return m
def __mul__(self, n):
'''乘法'''
# a * b (或：a * 2.0)
if isinstance(n, int) or isinstance(n,float):
m = matrix(self.row, self.column)
for r in range(self.row):
for c in range(self.column):
m[r, c] = self[r, c]*n
else:
assert n.row == self.column, "维度不匹配，不能相乘"
m = matrix(self.row, n.column)
for r in range(self.row):
for c in range(n.column):
sum = 0
for k in range(self.column):
sum += self[r, k] * n[k, r]
m[r, c] = sum
return m
def __p__(self, n):
'''除法'''
# a / b
pass
def __pow__(self, k):
'''乘方'''
# a**k
assert self.row == self.column, "不是方阵，不能乘方"
m = copy.deepcopy(self)
for i in range(k):
m = m * self
return m
def rank(self):
'''矩阵的秩'''
pass
def trace(self):
'''矩阵的迹'''
pass
def adjoint(self):
'''伴随矩阵'''
pass
def invert(self):
'''逆矩阵'''
assert self.row == self.column, "不是方阵"
m = matrix(self.row, self.column*2)
i = self.identity() # 单位矩阵
i.show()#############################
# 拼接
for r in range(1,m.row+1):
temp = self[r]
temp.extend(i[r])
m[r] = copy.deepcopy(temp)
m.show()#############################
# 初等行变换
for r in range(1, m.row+1):
# 本行首元素(m[r, r])若为 0，则向下交换最近的当前列元素非零的行
if m[r, r] == 0:
for rr in range(r+1, m.row+1):
if m[rr, r] != 0:
m[r],m[rr] = m[rr],m[r] # 交换两行
break
assert m[r, r] != 0, '矩阵不可逆'
# 本行首元素(m[r, r])化为 1
temp = m[r,r] # 缓存
for c in range(r, m.column+1):
m[r, c] /= temp
print("m[{0}, {1}] /= {2}".format(r,c,temp))
m.show()
# 本列上、下方的所有元素化为 0
for rr in range(1, m.row+1):
temp = m[rr, r] # 缓存
for c in range(r, m.column+1):
if rr == r:
continue
m[rr, c] -= temp * m[r, c]
print("m[{0}, {1}] -= {2} * m[{3}, {1}]".format(rr, c, temp,r))
m.show()
# 截取逆矩阵
n = matrix(self.row,self.column)
for r in range(1,self.row+1):
n[r] = m[r][self.row:]
return n
def jieti(self):
'''行简化阶梯矩阵'''
pass
def transpose(self):
'''转置'''
m = matrix(self.column, self.row)
for r in range(self.column):
for c in range(self.row):
m[r, c] = self[c, r]
return m
def cofactor(self, row, column):
'''代数余子式（用于行列式展开）'''
assert self.row == self.column, "不是方阵，无法计算代数余子式"
assert self.row >= 3, "至少是3*3阶方阵"
assert row <= self.row and column <= self.column, "下标超出范围"
m = matrix(self.column-1, self.row-1)
for r in range(self.row):
if r == row:
continue
for c in range(self.column):
if c == column:
continue
rr = r-1 if r > row else r
cc = c-1 if c > column else c
m[rr, cc] = self[r, c]
return m
def det(self):
'''计算行列式(determinant)'''
assert self.row == self.column,"非行列式，不能计算"
if self.shape == (2,2):
return self[1,1]*self[2,2]-self[1,2]*self[2,1]
else:
sum = 0.0
for c in range(self.column+1):
sum += (-1)**(c+1)*self[1,c]*self.cofactor(1,c).det()
return sum
def zeros(self):
'''全零矩阵'''
m = matrix(self.column, self.row, fill=0.0)
return m
def ones(self):
'''全1矩阵'''
m = matrix(self.column, self.row, fill=1.0)
return m
def identity(self):
'''单位矩阵'''
assert self.row == self.column, "非n*n矩阵，无单位矩阵"
m = matrix(self.column, self.row)
for r in range(self.row):
for c in range(self.column):
m[r, c] = 1.0 if r == c else 0.0
return m
def show(self):
'''打印矩阵'''
for r in range(self.row):
for c in range(self.column):
print(self[r+1, c+1],end=' ')
print()
if __name__ == '__main__':
m = matrix(3,3,fill=2.0)
n = matrix(3,3,fill=3.5)
m[1] = [1.,1.,2.]
m[2] = [1.,2.,1.]
m[3] = [2.,1.,1.]
p = m * n
q = m*2.1
r = m**3
#r.show()
#q.show()
#print(p[1,1])
#r = m.invert()
#s = r*m
print()
m.show()
print()
#r.show()
print()
#s.show()
print()
print(m.det())
以上就是python如何实现矩阵类的代码示例详解的详细内容。