python - 将列向量连接到矩阵的末尾

Question

我试图在矩阵的末尾添加一个列向量，如下所示：

import numpy as np
datas=[[1,2],[3,4]]
temp=[1,2]
datas=np.array(datas)
temp=np.transpose(np.array(temp))

np.append(datas,temp,axis=1)

但是我收到尺寸不匹配错误？

那我该怎么做呢？

标签： pythonnumpy

Answer 1

您需要向 temp 添加一维，以便两个数组具有相同的维度

import numpy as np
datas=[[1,2],[3,4]]
temp=[1,2]
datas=np.array(datas)
temp=np.array(temp)[:, np.newaxis] ## this adds new dimension 

np.append(datas,temp,axis=1)

你也可以使用如下的连接函数来完成。如果您连接两个以上的数组，它将表现更好。在这里，您在循环中创建 python list ls 然后将它们连接起来

ls = [datas,temp]
np.concatenate(ls, axis=1)