网上有很多关于科学计算包sympy的介绍,这里我把官方文档的英文表述贴过来。简单翻译就是sympy是个代数系统,底层完全使用python语言写的,使用简单、好理解、易扩展。
SymPy is a Python library for symbolic mathematics. It aims to become a full-featured computer algebra system (CAS) while keeping the code as simple as possible in order to be comprehensible and easily extensible. SymPy is written entirely in Python.
正好最近在研究线性代数,顺手把SymPy中关于矩阵的基本用法记录一下。
1、矩阵创建
矩阵的创建过程被理解为提供了一组行向量。 A matrix is constructed by providing a list of row vectors that make up the matrix.
import sympy
matrix_a = sympy.Matrix([[1, 3], [-2, 3]]) # 生成2*2的方阵
matrix_b = sympy.Matrix([[1, 0, 2], [2, 3, 4]]) # 生成2*3的矩形矩阵
2、创建列向量
list对象被理解为一个列向量。 a list of elements is considered to be a column vector.
column_vector_a = sympy.Matrix([1, 2, 1])
3、截取矩阵的某一行或列, 使用 row()和 col()
print(matrix_b.col(0)) # 打印第一列
print(matrix_b.row(0)) # 打印第一行
print(matrix_b.col(-1)) # 打印倒数第一列
print(matrix_b.row(-1)) # 打印倒数第一行
4、删除行向量、列向量
使用 row_del 或 col_del. 注意下:这些操作直接修改原来的矩阵对象,不会生成新的对象。
matrix_c = sympy.Matrix([[1, 0, 2, 4], [2, 3, 4, 0], [0, 1, 1, 1]])
matrix_c.row_del(0) # 删除第一行
matrix_c.col_del(1) # 删除第二列
5、插入行向量、列向量使用row_insert 或 col_insert。
insert注意三个点: (1)产生新的矩阵对象 (2)第一参数指的是插入位置,第二个参数是需要插入的内容 (3)注意区分插入行向量和列向量,Matrix()括号内略有不同,具体看下面的代码:
matrix_d = sympy.Matrix([[1, 0, -1], [-5, 3, 4]])
matrix_d_insert_col = matrix_d.col_insert(0, sympy.Matrix([8, 9])) # 在第一列插入列向量[8,9]
# print(matrix_d_insert_col)
matrix_d_insert_row = matrix_d.row_insert(0, sympy.Matrix([[1, 1, 1]])) # 在第一行插入行向量
# print(matrix_d_insert_row)
6、像加法、乘法这样的简单运算使用+,*, **就可以了。求逆矩阵,只需把power设置为-1即可。 simple operations like addition and multiplication are done just by using +, *, and **. To find the inverse of a matrix, just raise it to the -1 power.
"""矩阵乘法"""
matrix_multiplication = matrix_a * matrix_b
# print(matrix_multiplication)
"""矩阵求逆,如果不可逆,报错NonInvertibleMatrixError: Matrix det == 0; not invertible."""
matrix_inverse = matrix_a ** -1
# print(matrix_inverse)
"""矩阵转置"""
matrix_transpose = matrix_a.T
# print(matrix_transpose)
7、特殊矩阵的创建
"""生成单位矩阵,eye(n) will create an identity matrix"""
matrix_identity = sympy.eye(2) # 生成二阶单位矩阵
# print(matrix_identity)
"""生成n*m的零矩阵,zeros(m,n)"""
matrix_zero = sympy.zeros(2, 3) # 生成2*3的零矩阵
# print(matrix_zero)
"""生成元素都是1的矩阵,ones(m,n)"""
matrix_one = sympy.ones(2, 2) # 生成2*2的元素都为1的方阵
# print(matrix_one)
"""生成对角矩阵,diag(),参数可以是数字,也可以是矩阵
The arguments to diag can be either numbers or matrices.
A number is interpreted as a matrix. The matrices are stacked diagonally. """
matrix_diag_1 = sympy.diag(1, 2, 3) # 生成对角线元素为1,2,3的三阶方阵
# print(matrix_diag_1)
matrix_diag_2 = sympy.diag(1, sympy.ones(2, 2), sympy.Matrix([[1, 2], [1, 3]]))
# print(matrix_diag_2)