split an array into subarrays python
# For 2 arrays
x1, x2 = map(list, zip(*zip(*[iter(a)] * 2)))
print(x1, x2, sep='n')
[5, 3, 1]
[4, 2, 0]
# For 3 arrays
x1, x2, x3 = map(list, zip(*zip(*[iter(a)] * 3)))
print(x1, x2, x3, sep='n')
[5, 2]
[4, 1]
[3, 0]