matrix_exercism

class Matrix:
    def __init__(self, matrix_string):
        self.matrix_string = matrix_string

    def row(self, index):
        row = list(self.matrix_string.split("\n"))
        return row[index]

"""
My logic is: 
1. Take the string
2. Split at \n
3. Convert this into list(s) within a list
4. Access the required row with an index

What I have so far prints: 5 6 7 8
This fails the tests because it is expecting the output: [5, 6, 7, 8]

"""

    def column(self, index):
        pass


matrix = Matrix("1 2 3 4\n5 6 7 8\n9 8 7 6")
print(matrix.row(1))

I found a very cool solution in the community section. Now I work to understand it!!

class Matrix:
    def __init__(self, matrix_string):
        self.map = [[int(num) for num in rows.split()] for rows in matrix_string.split("\n")]

    def row(self, index):
        return self.map[index-1]

    def column(self, index):
        return [row[index-1] for row in self.map]