from itertools import count
import numpy as np

def get_subarrays(x): # provide additional parameters, e.g. start, step etc.
    for s in (x[j:i, :] for j, i in zip(count(0, 540), count(480, 540))):
        if s.size:
            yield s
        else:
            raise StopIteration

x = np.random.rand(10000, 10)
data = np.concatenate(list(get_subarrays(x)))
print(data.shape)

from itertools import count
# count returns infinite generator, e.g. count(10, 5) creates generator starting at 10 with step 5: 10, #15, 20, ... this sequence never ends
# So, if you execute
# for j in count(10, 5): # this is infinite loop
#     print(j) 
# that execution will never been stopped, until, e.g.,  Ctrl+C.

import numpy as np

# (x[j:i, :] for j, i in zip(count(0, 540), count(480, 540))) is a generator:
# it produces x[0:480, :], x[540:480+540, :], x[1080:1080+540, :] etc. 
# when the index becomes greater than the size of x, this generator returns empty numpy array, 
# This generator will never stop. To stop this generator we use for-loop and check size of returned
# subarray (s) on each iteration. If returned subarray becomes empty we raise the StopIteration exception.

# get_subarrays is a generator, that extracts subarrays from source array x
def get_subarrays(x):
    for s in (x[j:i, :] for j, i in zip(count(0, 540), count(480, 540))):
        if s.size:
            yield s
        else:
            raise StopIteration
 
x = np.random.rand(10000, 10)  # Test array

# We need to pass a list of arrays to be concatenated, so, lets create such list. 

#list(get_subarrays(x)) is equivalent for 
#res = []
#for item in get_subarrays(x): # This loop is break, when StopIteration is raised (this is common behavior for Python generators/iterators and loops)
#    res.append(item)

# Now, we can pass `res` to np.concatenate, or, for short, use list(get_subarrays(x)) instead of `res`.
data = np.concatenate(list(get_subarrays(x)))

print(data.shape)

for j, i, k in zip(count(0, 540), count(480, 540), range(10)):
    print(k, j, ':', i)

Output:
0 0 : 480
1 540 : 1020
2 1080 : 1560
3 1620 : 2100
4 2160 : 2640
5 2700 : 3180
6 3240 : 3720
7 3780 : 4260
8 4320 : 4800
9 4860 : 5340