PracticeDataStructuresArrays2DArray - DS.py

import itertools
# 6 X 6 Metrics
list2d = [[1, 2, 3, 4, 5, 6], #head only
            [1, 1, 1, 0, 0, 0], #head only and body
            [1, 1, 1, 0, 0, 0], #both head , tail and body
            [0, 0, 2, 4, 4, 0], #both head , tail and body
            [0, 0, 0, 2, 0, 0], #tail and body
            [0, 0, 1, 2, 4, 0]] #tail only
# check if the metrics is square oterwise we're not playing the glass game, we might be able to but probably square is easier        
def squaremetrix(arr):
    listrows = len(arr) #number of rows
    listcols = len(arr[0]) #number of columns selecting the first row and counting it's 
    for i in arr:
        if len(i) != listcols or listcols !=listrows and i == arr[-1]:
            return "list is not a square"
        elif len(i) == listcols and i == arr[-1]:
            # print("list is a square")
            return "list is a square", listrows , listcols
        else:
            pass
    
isitsqare = squaremetrix(list2d)

def glasshours(arr):
    if squaremetrix(list2d)[0] == 'list is a square':
        matrixShape = [squaremetrix(list2d)[1] , squaremetrix(list2d)[1]]
        # print("matrix shape ",matrixShape)
        
        for index, i in enumerate (arr):
            lengthofi = len(i)
            howmanyheads = int(lengthofi/3) + 2 #length of row divide by head size + 2
            headsNtailes = list(itertools.repeat(i, howmanyheads)) #repeat the same item the number of heads we need
            counter = -1
            #heads
            for j in headsNtailes:
                print(j[counter+1:counter+4])
                counter += 1
            #bodys
            for index2, k in enumerate (headsNtailes):
                if index != 0 or -1 and index2 != 0 or -1:
                    print (k)

                     

glasshours(list2d)