Numpy tömb nem egyenlő előre megadott részekre vágása hatékonyan?

Van ez a két tömböm: A = [ 0 1 4 2 3 7 5 6 9 12 13 8 11 14 10]

B = [3 3 5 3 1]

Az A tömböt szeretném felbontani a B tömbben szereplő méretekre, úgy hogy az első 3 elem kerüljön, egy tömb-be, majd a következő 3 elem is külön tömb-be, majd a következő 5 elem is külön tömb-be és így tovább. Tehát valahogy így nézne ki a végeredmény: [0,1,4],[2,3,7],[5,6,9,12,13],[8,11,14],[10]

Ezt meg lehet valahogy valósítani csak tisztán Numpy-ban, hatékonyan?