矩阵运算基本封装

在学习 Numpy 之前，想着自己去封装一些个关于矩阵的方法，平时工作比较忙，时间不多，还没写完，下周继续，话不多说，直接贴代码：

#!/usr/local/bin/python3
# coding: utf-8
import math


class Matrix:
    """矩阵构造类"""
    def __init__(self):
        pass

    @staticmethod
    def create(array, row=1, col=0):
        """
        创建矩阵
        :param array: 用于创建矩阵的一维列表
        :param row: (可选参数) 矩阵行数
        :param col: (可选参数) 矩阵列数
        """
        matrix = []
        if Matrix.is_list(array):
            matrix = array
        _length = len(array)
        rows = row
        cols = max(math.ceil(_length // rows), col)
        _i, _j = 0, 0
        while _i < rows:
            _row = list(array[_j:_j + cols])
            while len(_row) < cols:
                _row.append(0)
            matrix.append(_row)
            _i += 1
            _j += cols
        return matrix

    @staticmethod
    def each(matrix, callback):
        """
        矩阵遍历
        :param matrix: 矩阵
        :param callback: 回调函数, 接受三个参数: 当前值, 当前行, 当前列
        """
        range_row = range(matrix.rows)
        range_col = range(matrix.cols)
        for _i in range_row:
            for _j in range_col:
                callback(matrix.matrix[_i][_j], _i, _j)

    @staticmethod
    def flat(matrix):
        """
        矩阵扁平化
        :return array: 扁平化后的列表
        """
        array = []
        Matrix.each(matrix, lambda _v, _i, _j: array.append(_v))
        return array

    @staticmethod
    def add(matrix1, matrix2):
        """
        矩阵相加
        :param matrix1: 需要相加的矩阵
        :param matrix2: 需要加上的矩阵
        :return: 相加后的矩阵
        """
        def _add(_v, _i, _j):
            matrix1[_i][_j] += _v
        Matrix.each(matrix2, _add)
        return matrix1

    @staticmethod
    def is_list(array):
        """判断是否为列表"""
        return isinstance(array, list)

    @staticmethod
    def is_square(matrix):
        """判断是否为对称矩阵"""
        if Matrix.is_list(matrix):
            row_count = len(matrix)
            for _row in matrix:
                if Matrix.is_list(_row):
                    if not len(_row) == row_count:
                        return False
                    else:
                        pass
                else:
                    return False
        else:
            return False
        return True

    @staticmethod
    def size(matrix):
        """返回矩阵维度"""
        return len(matrix), len(matrix[0])


def main():
    mt = Matrix.create(range(1, 20, 2), 3, 3)
    mt1 = Matrix.create(range(2, 30, 2), 4, 3)
    res = Matrix.is_square(mt1)
    print(res)


if __name__ == '__main__':
    main()

下周把这一块做完。