numpy 如何获取Numy数组的结尾和开头

46qrfjad 于 2022-11-10 发布在其他

关注(0)|答案(1)|浏览(167)

我想要一个麻木数组的结尾和开头。
示例：

a = np.array([1,2,3,4,5])
print(a[-2:2])

我所期待的：[4,5,1]
但实际上它给出了一个空数组[]。
这在某种程度上可行吗？

1条答案

以下是一种方法
按指定的计数滚动数组，以便标题在结束后滚动，然后对数组进行切片

a = np.array([1,2,3,4,5])

np.roll(a, -1)[2:]

或

a = np.array([1,2,3,4,5])
values_from_start=1
no_of_values = 3

start = len(a) - no_of_values

np.roll(a, values_from_start*(-1) )[start:]

array([4, 5, 1])