2024-10-04 05:31:46 发布
网友
我有一个向量,我想选择所有奇数编号的索引项与向量的第一个和最后一个项并集。例如,让向量:
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]
它是有效的,但是,有没有一种更像Python的或干净的方法来做同样的事情呢
一种替代方法是使用np.r_构建索引,如下所示:
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])
一种替代方法是使用
np.r_
构建索引,如下所示:相关问题 更多 >
编程相关推荐