python - 如何使 2d numpy 数组成为 3d 数组？

Question

我有一个形状为 (x, y) 的 2d 数组，我想将其转换为形状为 (x, y, 1) 的 3d 数组。有没有很好的 Pythonic 方式来做到这一点？

Answer 1

除了其他答案之外，您还可以将切片与 numpy.newaxis 一起使用:

>>> from numpy import zeros, newaxis
>>> a = zeros((6, 8))
>>> a.shape
(6, 8)
>>> b = a[:, :, newaxis]
>>> b.shape
(6, 8, 1)

甚至是这个(适用于任意数量的维度):

>>> b = a[..., newaxis]
>>> b.shape
(6, 8, 1)