将形状（n，1）的numpy数组转换为形状（n，）[重复]

Question

我想将一个形状为 (n,1) 的 numpy 数组转换为 (n,)。是否可以在不必遍历数组的情况下做到这一点？

Answer 1

使用numpy.squeeze:

>>> x = np.array([[[0], [1], [2]]])
>>> x.shape
(1, 3, 1)
>>> np.squeeze(x).shape
(3,)
>>> np.squeeze(x, axis=(2,)).shape
(1, 3)

我很确定您可以通过 Google 搜索找到此答案。 . .