matrix exercise

Given a string representing a matrix of numbers, return the rows and columns of that matrix.

So given a string with embedded newlines like:
"1 2 3\n4 5 6\n7 8 9\n8 7 6"

your code should be able to spit out:

A list of the rows, reading each row left-to-right while moving top-to-bottom across the rows,
A list of the columns, reading each column top-to-bottom while moving from left-to-right.
The rows for our example matrix:

9, 8, 7
5, 3, 2
6, 6, 7
And its columns:

9, 5, 6
8, 3, 6
7, 2, 7
In this exercise you're going to create a class. Don't worry, it's not as complicated as you think!


class Matrix(object):

    def __init__(self, string):
        string = string.splitlines()
        self.setMatrix = [list(map(int,i.split())) for i in string]

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

    def column(self, index):
        col = [i[index-1] for i in self.setMatrix]
        return col