如何使 2d numpy 数组成为 3d 数组? [英] How to make a 2d numpy array a 3d array?

问题描述

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

解决方案

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

<预><代码>>>>从 numpy 导入零点，newaxis>>>a = 零 ((6, 8))>>>一个形状(6, 8)>>>b = a[:, :, newaxis]>>>b.形状(6, 8, 1)

甚至这个(它可以处理任意数量的维度):

<预><代码>>>>b = a[..., newaxis]>>>b.形状(6, 8, 1)

I have a 2d array with shape (x, y) which I want to convert to a 3d array with shape (x, y, 1). Is there a nice Pythonic way to do this?

解决方案

In addition to the other answers, you can also use slicing with numpy.newaxis:

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

Or even this (which will work with an arbitrary number of dimensions):

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

这篇关于如何使 2d numpy 数组成为 3d 数组?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！