【PLA】基于Python实现的线性代数算法库之矩阵

【PLA】基于Python实现的线性代数算法库之矩阵

算法包下载链接：https://download.csdn.net/download/qq_42629529/79481514

from PLA.Vector import Vector
from PLA.Matrix import Matrix

if __name__ == "__main__":

    matrix = Matrix([[1, 2], [3, 4]])
    print(matrix)
    #获得行列数
    print("matrix.shape = {}".format(matrix.shape()))
    #获得元素个数
    print("matrix.size = {}".format(matrix.size()))
    #行数
    print("len(matrix) = {}".format(len(matrix)))
    #行列对应元素
    print("matrix[0][0] = {}".format(matrix[0, 0]))

    matrix2 = Matrix([[5, 6], [7, 8]])
    print(matrix2)
    #加
    print("add: {}".format(matrix + matrix2))
    #减
    print("subtract: {}".format(matrix - matrix2))
    #数乘
    print("scalar-mul: {}".format(2 * matrix))
    #乘数
    print("scalar-mul: {}".format(matrix * 2))
    #0矩阵
    print("zero_2_3: {}".format(Matrix.zero(2, 3)))

    T = Matrix([[1.5, 0], [0, 2]])
    p = Vector([5, 3])
    #乘法
    print("T.dot(p) = {}".format(T.dot(p)))

    P = Matrix([[0, 4, 5], [0, 0, 3]])
    print("T.dot(P) = {}".format(T.dot(P)))

    print("A.dot(B) = {}".format(matrix.dot(matrix2)))
    print("B.dot(A) = {}".format(matrix2.dot(matrix)))
    #转置
    print("P.T = {}".format(P.T()))
    #单位矩阵
    I = Matrix.identity(2)
    print(I)
    print("A.dot(I) = {}".format(matrix.dot(I)))
    print("I.dot(A) = {}".format(I.dot(matrix)))