How to add an extra column to a NumPy array

前端 未结 17 1922
一个人的身影
一个人的身影 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:24

    Use numpy.append:

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

提交回复
热议问题