Python数据限制的简写

Question

给定一个numpy数组:

a = arange(10,20,1)

我经常需要一个包含数组的第一个和最后一个元素的元组:

w = a[0],a[-1]

是否有一个方便的python切片快捷方式在单个引用中执行此操作a？

Answer 1

是的,使用Numpy的高级索引:

w = a[[0, -1]]