How to add an extra column to a NumPy array

前端 未结 17 2148
一个人的身影
一个人的身影 2020-11-22 14:37

Let’s say I have a NumPy array, a:

a = np.array([
    [1, 2, 3],
    [2, 3, 4]
    ])

And I would like to add a column of ze

17条回答
  •  甜味超标
    2020-11-22 15:32

    np.concatenate also works

    >>> a = np.array([[1,2,3],[2,3,4]])
    >>> a
    array([[1, 2, 3],
           [2, 3, 4]])
    >>> z = np.zeros((2,1))
    >>> z
    array([[ 0.],
           [ 0.]])
    >>> np.concatenate((a, z), axis=1)
    array([[ 1.,  2.,  3.,  0.],
           [ 2.,  3.,  4.,  0.]])
    

提交回复
热议问题