python中的维度_访问python中的第n个维度

如果您想要一个视图并希望它快速,您可以手动创建索引：

arr[(slice(None), )*5 + (your_index, )]

# ^---- This is equivalent to 5 colons: `:, :, :, :, :`

这比np.take快得多,只比索引慢一点：s：

import numpy as np

arr = np.random.random((10, 10, 10, 10, 10, 10, 10))

np.testing.assert_array_equal(arr[:,:,:,:,:,4], arr.take(4, axis=5))

np.testing.assert_array_equal(arr[:,:,:,:,:,4], arr[(slice(None), )*5 + (4, )])

%timeit arr.take(4, axis=5)

# 18.6 ms ± 249 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)

%timeit arr[(slice(None), )*5 + (4, )]

# 2.72 µs ± 39.7 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

%timeit arr[:, :, :, :, :, 4]

# 2.29 µs ± 107 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

但也许不是可读的,所以如果你经常需要它,你可能应该把它放在一个有意义的名字的函数中：

def index_axis(arr, index, axis):

return arr[(slice(None), )*axis + (index, )]

np.testing.assert_array_equal(arr[:,:,:,:,:,4], index_axis(arr, 4, axis=5))

%timeit index_axis(arr, 4, axis=5)

# 3.79 µs ± 127 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)