finding every squares value in array

July 31, 2022

I have different sizes of arrays that each element is its index if it was flatten. Is there a way to print out every element per square going clockwise? I thought about slicing the arrays but that doesn’t go clockwise and only prints one square and not all.

arr1 = np.array([[0, 1],[2, 3]])
arr2 = np.array([[0, 1, 2],[3, 4, 5]])
arr3 = np.array([[0, 1],[2, 3],[4, 5]])

print(arr1[0:2,0:2])
print()
print(arr2[0:2,0:2])
print()
print(arr3[0:2,0:2])

output:

[[0 1]
 [2 3]]

[[0 1]
 [3 4]]

[[0 1]
 [2 3]]

>Solution :

Maybe this helps

import numpy as np


a = np.random.randint(0, 10, size=(7, 9))

print(a)


for i in range(a.shape[0]):
    for j in range(a.shape[1]):
        x = a[i:i+2, j:j+2]
        if x.flatten().size == 4:
            print(x) # every 2 by 2 array of 4 elements