圆形numpy数组索引

2024-04-24 04:11:42 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个一维numpy数组a = [1,2,3,4,5,6],还有一个函数,它得到两个输入starting_indexending_index,并返回a[staring_index:ending_index]

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

例如,如果starting_index=4ending_index=1,那么输出应该是output = [5,6,1]。我可以用if条件实现它,但我想知道是否有任何Pythonic和简洁的方法来实现它?


Tags: 函数numpy元素outputindexifending方式