将元素追加到numpy数组中每个箭头的末尾

Question

如果我有一个像numpy数组：

x= [[3, 3], [2, 2]]

我想在每个行的末尾添加一个元素-1，如下所示：

x= [[3, 3, -1], [2, 2, -1]]

任何简单的方法吗？

Answer 1

一种简单的方法是使用np.insert -

np.insert(x,x.shape[1],-1,axis=1)

我们也可以使用np.column_stack -

np.column_stack((x,[-1]*x.shape[0]))

示例运行 -

In [161]: x
Out[161]: 
array([[0, 8, 7, 0, 1],
       [0, 1, 8, 6, 8],
       [3, 4, 7, 0, 2]])

In [162]: np.insert(x,x.shape[1],-1,axis=1)
Out[162]: 
array([[ 0,  8,  7,  0,  1, -1],
       [ 0,  1,  8,  6,  8, -1],
       [ 3,  4,  7,  0,  2, -1]])

In [163]: np.column_stack((x,[-1]*x.shape[0]))
Out[163]: 
array([[ 0,  8,  7,  0,  1, -1],
       [ 0,  1,  8,  6,  8, -1],
       [ 3,  4,  7,  0,  2, -1]])