Tag: algorithm

I often use MathJax and often have to write matrices. Since writing matrices with MathJax is very tedious, I opted for automation.

Code:

import numpy as np
import pyperclip


class Conversion():


    def __init__(self, matrix, n = None, m = None):
        if matrix == None:
            self.matrix = np.random.randint(-10,10, size = (n,m))
        else:
            '''If variable "matrix", for example, contains (1,2,3,4), instead of ((1,2,3,4)), functions below 
            will throw error, hence we need to make sure that the data type is correct.'''
            if type(matrix(0)) in (str,int,float):
                self.matrix = np.array((matrix,))
            else:
                self.matrix = np.array(matrix)

        self.rows = len(self.matrix)
        self.cols = len(self.matrix(0))


    def single(self, choosen_brackets = ')'):
        available_brackets = {' ' : 'matrix', 
                              ')' : 'pmatrix', 
                              ')' : 'bmatrix', 
                              '}' : 'Bmatrix', 
                              '|' : 'vmatrix',
                              '||': 'Vmatrix'  }
        choosen_brackets = available_brackets(choosen_brackets)
        body = '$$\begin {} n'.format('{' + choosen_brackets + '}')
        for row in self.matrix: 
            row = (str(x) for x in row)
            body += ' & '.join(row) + '\\' + 'n'
        body +='end {} $$'.format('{' + choosen_brackets + '}')
        print(body)
        pyperclip.copy(body)


    def augmented(self, choosen_brackets = '()'):
        '''We are assuming that the last column of the given matrix is a vector.'''

        pos_of_the_verticar_bar = '{' + 'c'*(self.cols-1) + '|' + 'c' + '}'
        body = '$$\left ( \begin {array} %s n' % (pos_of_the_verticar_bar)
        for row in self.matrix:
            row = (str(x) for x in row)
            body += ' & '.join(row) + '\\' + 'n'
        body +='end {array} \right )$$'
        print(body)
        pyperclip.copy(body)

Remarks:

1. There work augmented is pretty much too singleI could have merged them into one. However, I think that the separation of the codes improves readability.

2. MSE includes Mathjax equations $$ Instead of $

3. "single"is a bad name for a function, I admit, I have not found any better options, do not hesitate to offer suggestions.

What can be improved?