连接两个numpy数组

Question

我编写了以下代码来附加两个numpy数组：

td_XN = searchNegative(X,y,10)
td_XP = searchPosotive(X,y,10)
print(np.array(td_XN).shape, np.array(td_XP).shape)
print(type(td_XN), type(td_XP))
td_X = np.concatenate(td_XP, td_XN)
td_y = [1,1,1,1,1,1,1,1,1,1,0,0,0,0,0,0,0,0,0,0]
print(td_X.shape, len(td_y))

但是，它会产生以下错误：

^{pr2}$

在这条线上：

td_X = np.concatenate(td_XP, td_XN)

Answer 1

如果您想并排连接（也就是说，创建一个10×2544*2的数组）：可以这样做

td_X = np.concatenate([td_XP, td_XN],axis=1)

例如

^{pr2}$

给予

array([[1, 2, 3, 7, 1, 2, 3],
       [4, 5, 6, 8, 4, 5, 6]])

另一方面，如果您想在td_XP下面添加td_XN，您可以这样做

td_X = np.concatenate([td_XP, td_XN],axis=0)

例如

td_X = np.concatenate([[[1,2,3],[4,5,6]],[[1,2,7],[4,5,8]]],axis=0)

给予

array([[1, 2, 3],
       [4, 5, 6],
       [1, 2, 7],
       [4, 5, 8]])