add column and row accessors

Signed-off-by: Csaba Kiraly <csaba.kiraly@gmail.com>
This commit is contained in:
Csaba Kiraly 2022-12-07 15:10:35 +01:00
parent 13580cd44d
commit 8da447ac5a
No known key found for this signature in database
GPG Key ID: 0FE274EE8C95166E

View File

@ -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)