我想把一个numpy数组添加到另一个numpy数组的每一行末尾。

Question

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

如果我有numpy数组。

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

在每一行的末尾，我想添加另一个numpy数组。([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]])

请记住这些数组的形状是不同的.

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))