class Matrix: def __init__(self, matrix_string): self.rows = [[int(s) for s in row_string.split()] for row_string in matrix_string.split('\n')] def row(self, index): if index < 1: raise ValueError("Row index must be no less than 1") return self.rows[index - 1] def column(self, index): if index < 1: raise ValueError("Row index must be no less than 1") return [row[index - 1] for row in self.rows]