索引在numpy ndarray

Question

我有一个numpy nd数组，其形状如下：

import numpy as np

# Note this can be variable and have different dimensions each time
shape = [100, 100, 100, 5]
array = np.zeros(shape)

现在我要做的是解决最后一个维度，所以我需要这样的东西：

array[:, :, :, i] = something

但是我不能像这样硬编码，因为尺寸可以根据输入而改变。那么有没有办法索引nd-array的最后一个维度？

Answer 1

烨：

array[..., i] = something

那是一个文字...，而不是你想要写的其他东西的占位符。