如何在数组numpy python

时间:2017-04-05 15:25:04

标签: python arrays numpy

是否有任何函数可以让我在数组的每一行末尾添加任何数字(例如" 0")。 例: 我有二维数组:

ar=[[0,0,1],
[1,1,1],
[0,1,0]]

我想将它添加到其他1维数组中,所以我有:

 otherarray=numpy.array([],dtype=bool)
    otherarray=np.append(otherarray, ar)

结果:

 otherarray=[0,0,1,1,1,1,0,1,0]

它有效。但我需要在任何数字的每一行添加,例如0并在otherarray(不修改ar)中得到它。 结果我想要:

[0,0,1,0,1,1,1,0,0,1,0,0]

我正在使用for循环(我将每个元素逐个地放入其中)但现在我问:是否有更好的方法?

1 个答案:

答案 0 :(得分:1)

您可以将零列附加到ar,然后展平它:

A = np.array(ar)    

np.hstack([A, np.zeros((A.shape[0], 1), dtype=A.dtype)]).ravel()
# array([0, 0, 1, 0, 1, 1, 1, 0, 0, 1, 0, 0])