diff --git a/DAS/block.py b/DAS/block.py index 5539865..5c09ca1 100644 --- a/DAS/block.py +++ b/DAS/block.py @@ -15,6 +15,18 @@ class Block: for i in range(self.blockSize*self.blockSize): self.data[i] = random.randint(1, 9) + def getColumn(self, columnID): + column = [0] * self.blockSize + for i in range(self.blockSize): + column[i] = self.data[(i*self.blockSize)+columnID] + return column + + def getRow(self, rowID): + row = [0] * self.blockSize + for i in range(self.blockSize): + row[i] = self.data[(rowID*self.blockSize)+i] + return row + def print(self): dash = "-" * (self.blockSize+2) print(dash)