Python/Numpy - 在数组末尾包裹切片

Question

我有两个1D数组,一个有一些感兴趣的值(a),另一个为该数组提供索引(b).我知道b中的值总是增加,除了在数字减少的一个点(可能是任何地方),因为它从数组a的末尾滚动到数组的开头.下面的方法似乎有效,但我认为必须采用更清洁的方式.谁能提出更好的建议？谢谢.

码:

import numpy as np
a = np.arange(12)
b = np.array([5, 9, 2, 4])
#I want to generate these:
#[5,6,7,8,9]
#[9,10,11,0,1,2]
#[2,3,4]
#[4,5]

a = np.roll(a, -b[0], axis=0)
# Subtract off b[0] but ensure that all values are positive
b = (b-b[0]+len(a))%len(a)
for i, ind in enumerate(b):
   if i < len(b)-1:
      print a[b[i]:b[i+1]+1]
   else:
      print np.hstack((a[b[i]:len(a)], a[0]))

Answer 1

有点短，但也许我还能做得更好......

import numpy as np

a = np.arange(12)
b = np.array([5, 9, 2, 4])
b = np.append(b, b[0])

for i in range(0, len(b)-1):
    print np.roll(a, len(a)-b[i])[:b[i+1]-b[i]+1]