python - 我想在另一个 numpy 数组的每一行的末尾添加一个 numpy 数组

Question

我想组合两个 numpy 数组。np.concatenate/vstack/hstack只将它添加到数组的末尾，但我想将它添加到数组中每一行的末尾。

如果我有 numpy 数组，

([[1,3,1],
[2,5,3],
 [4,8,4]])

在每一行的末尾，我想添加另一个 numpy array ([8,9,5,5,3,1])。

所以最后最终的数组看起来像这样：

([[1,3,1,8,9,5,5,3,1],
[2,5,3,8,9,5,5,3,1],
[4,8,4,8,9,5,5,3,1]])

请记住，这些确实有不同的形状。

标签： pythonnumpynumpy-ndarray

Answer 1

希望对你有帮助。

import numpy as np

a = np.array([[1,3,1],[2,5,3], [4,8,4]])
b = np.array([8,9,5,5,3,1])
b = b[None,:].repeat(3,axis=0)

print(np.concatenate((a,b),axis=1))