我有一个numpy数组A
,如下:
A.shape[axis] = n+1.
现在我想通过沿着轴axis
选择索引0,..,n-1
和1,...,n
分别构建两个切片B
和C
,如下:
B.shape[axis] = C.shape[axis] = n
B
和C
在其他轴上的大小与A
相同。不得复制数据。
# exemple data
A = np.random.rand(2, 3, 4, 5)
axis = 2
n = A.ndim
# building n-dimensional slice
s = [slice(None), ] * n
s[axis] = slice(0, n - 1)
B = A[s]
s[axis] = slice(1, n)
C = A[s]
一行代码:
B = A[[slice(None) if i != axis else slice(0, n-1) for i in xrange(n)]]
C = A[[slice(None) if i != axis else slice(1, n) for i in xrange(n)]]