对numpy数组的n维感到困惑？

Question

我有一个5-d numpy数组，形状是（5，1000，32，32，3），这意味着有3个32 * 32像素的通道，以及1000个样本，5个不同的时间戳。如何打印特定的32 * 32数据，例如，我要从16个样本，第二个时间戳，第一个通道打印32 * 32个数据？

Answer 1

可以将inedexing和slicing混合在一起，如下所示：

arr = np.random.randint(1000, size=(5, 1000, 32, 32, 3))
result = arr[1, 15, :, :, 0]
print(result.shape)

这将输出结果的形状：

  

（32，32）