MicroPython Forum (Archive)

Code: Select all

def add(a, b):
    return a + b


def sub(a, b):
    return a - b


def mul(a, b):
    return a * b


class Matrix(object):

    def __init__(self, mat):
        self.mat = mat

    def __repr__(self):
        return self.mat

    def __add__(self, b):
        return Matrix([list(map(add, *ab)) for ab in zip(self.mat, b.mat)])

    def __sub__(self, b):
        return Matrix([list(map(sub, *ab)) for ab in zip(self.mat, b.mat)])

    def __matmul__(self, b):
        return Matrix([[sum(map(mul, _a, _b)) for _b in zip(*b.mat)] for _a in self.mat])

Code: Select all

>>> a = Matrix([[1, 2, 3], [4, 5, 6]])
>>> b = Matrix([[10, 11], [20, 21], [30, 31]])
>>> c = Matrix([[-23, 3, 8], [-13, 17, -21]])
>>> d = Matrix([[32, 17], [-3, 11]])
>>> a @ b
[[140, 146], [320, 335]]
>>> a + c
[[-22, 5, 11], [-9, 22, -15]]
>>> a - c
[[24, -1, -5], [17, -12, 27]]
>>> a @ b + d
[[172, 163], [317, 346]]

Code: Select all

def add(a, b):
    return a + b


def sub(a, b):
    return a - b


def mul(a, b):
    return a * b


def square(a):
    return a * a


def eq(a, b):
    return a == b


class Matrix(object):

    def __init__(self, mat):
        """Matrix A, list of list"""
        self.mat = mat

    def dim(self):
        """Matrix dimension m rows by n cols"""
        return len(self.mat), len(self.mat[0])

    def map(self, func):
        """Map function to all elements of the matrix"""
        return Matrix([list(map(func, m)) for m in self.mat])

    def sumsquare(self):
        """Sum square of elements"""
        return sum(sum(m) for m in self.actfunc(square).mat)

    def meansquare(self):
        """Mean square of elements"""
        return self.sumsquare() / mul(*self.dim())

    def t(self):
        """Matrix transpose"""
        return Matrix(list(zip(*self.mat)))

    def __str__(self):
        return '{}'.format(self.mat)

    def __repr__(self):
        return '{}'.format(self.mat)

    def __neg__(self):
        """- A"""
        return self.map(lambda e: -e)

    def __pos__(self):
        """+ A"""
        return self.map(lambda e: e)

    def __add__(self, other):
        """A + B"""
        if not isinstance(other, Matrix):
            return Matrix([[_n + other for _n in _m] for _m in self.mat])
        return Matrix([list(map(add, *ab)) for ab in zip(self.mat, other.mat)])

    def __radd__(self, other):
        """B + A"""
        return self.__add__(other)

    def __iadd__(self, other):
        """A += B"""
        self.mat = self + other
        return self.mat
 # ....

Code: Select all

from matrix import Matrix

a = Matrix([[1, 2, 3], [4, 5, 6]])
b = Matrix([[10, 11], [20, 21], [30, 31]])
c = Matrix([[-23, 3, 8], [-13, 17, -21]])
d = Matrix([[32, 17], [-3, 11]])
e = Matrix([[9, 8, 7], [6, 5, 4]])

print('a + c =', a + c)
print('a - c =', a - c)

print('a + 2 =', a + 2)
print('2 + a =', 2 + a)  # TypeError: unsupported types for __add__: 'int', 'Matrix'