將形狀（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) 

我敢肯定，你可以找到從谷歌搜索這個答案。 。 。