python - 圆形numpy数组索引

我有一个一维numpy数组a = [1,2,3,4,5,6]和一个函数，该函数获取两个输入starting_index和ending_index，并返回a[staring_index:ending_index]。

显然，当ending_index小于starting_index时，我会遇到麻烦。在这种情况下，该函数应从starting_index开始，并以循环方式遍历向量a，即返回starting_index之后的所有元素以及所有从索引0到ending_index的元素。

例如，如果为starting_index=4和ending_index=1，则输出应为output = [5,6,1]。我可以用if条件实现它，但是我想知道是否有任何Pythonic简洁的方法可以做到这一点？

最佳答案

np.take具有wrap模式:

In [171]: np.take(np.arange(1,7),range(4,7),mode='wrap')
Out[171]: array([5, 6, 1])

那不是您想要的。

实际上，模量做同样的事情

In [177]: a[np.array([4,5,6])%6]
Out[177]: array([5, 6, 1])

但是，将(4,1)转换为[4, 5, 6]的小函数呢？或者，如果您更喜欢[4, 5, 0]呢？

def foo(a, start, stop):
    # fn to convert your start stop to a wrapped range
    if stop<=start:
        stop += len(a)
    return np.arange(start, stop)%len(a)

a[foo(a,4,1)]  # or
np.take(a,foo(a,4,1))

关于python - 圆形numpy数组索引，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/28398220/

NumPy数组

python - 圆形numpy数组索引