在 Python 中创建垂直 NumPy 数组

我在 Python 中使用 NumPy 来处理数组。这是我用来创建垂直数组的方法：

import numpy as np
a = np.array([[1],[2],[3]])

有没有一种简单、更直接的方法来创建垂直数组？

您可以使用reshape or vstack http://docs.scipy.org/doc/numpy/reference/generated/numpy.vstack.html :

>>> a=np.arange(1,4)
>>> a
array([1, 2, 3])
>>> a.reshape(3,1)
array([[1],
       [2],
       [3]])
>>> np.vstack(a)
array([[1],
       [2],
       [3]])

另外，您可以使用广播 http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html为了重塑你的数组：

In [32]: a = np.arange(10)
In [33]: a
Out[33]: array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

In [34]: a[:,None]
Out[34]: 
array([[0],
       [1],
       [2],
       [3],
       [4],
       [5],
       [6],
       [7],
       [8],
       [9]])