使用两个标准的花式索引

时间:2017-11-10 16:40:23

标签: python numpy

我有一个向量,我想选择所有奇数编号的索引条目与向量的第一个和最后一个条目联合。例如,让vector:

v = np.array([0,1,2,3,4,5,6])

我想做一些返回元素的选择:

v[someindexing]

返回

(0,1,3,5,6)

我这样做了:

mask = [0] + list(range(1,len(x)-1,2)) + [-1]
x[mask]

它有效,但是,有更多的pythonic或干净的方法来做同样的事情吗?

1 个答案:

答案 0 :(得分:2)

另一种方法是使用np.r_构建索引:

np.r_[0,1:len(v)-1:2,-1]
v = np.array([0,1,2,3,4,5,6])

v[np.r_[0,1:len(v)-1:2,-1]]
# array([0, 1, 3, 5, 6])