Preserving The Dimensions Of A Slice From A Numpy 3d Array
I have a 3d array, a, of shape say a.shape = (10, 10, 10) When slicing, the dimensions are squeezed automatically i.e. a[:,:,5].shape = (10, 10) I'd like to preserve the number of
Solution 1:
a[:,:,[5]].shape
# (10,10,1)
a[:,:,5]
is an example of basic slicing.
a[:,:,[5]]
is an example of integer array indexing -- combined with basic slicing. When using integer array indexing the resultant shape is always "identical to the (broadcast) indexing array shapes". Since [5]
(as an array) has shape (1,)
,
a[:,:,[5]]
ends up having shape (10,10,1)
.
Post a Comment for "Preserving The Dimensions Of A Slice From A Numpy 3d Array"