如何在Python中获得数组的子形状？

Question

不确定标题是否正确，但是我有一个形状为（84,84,3）的数组，我需要获取该形状为（84,84）的数组的子集，不包括第三维。

如何使用Python完成此操作？

Answer 1

your_array[:,:,0]

这称为切片。这个特定的示例获取数组的第一个“层”。假设您的子形状是单层。

Answer 2

如果使用的是numpy数组，则使用slices将是一种标准的方法：

import numpy as np

n = 3  # or any other positive integer
a = np.empty((84, 84, n))

i = 0  # i in [0, n]
b = a[:, :, i]

print(b.shape)

我建议您看看this。