在二维数组的每行末尾附加一个列表

2023-12-24

我想在二维数组 (a) 的每行末尾附加一个列表/一维数组 (b)

input:

a = np.array([[1, 1], [2, 2], [3, 3]])
b = np.array([4, 4])

期望输出:

array([[1, 1, 4, 4],
       [2, 2, 4, 4],
       [3, 3, 4, 4]])

my code:

temp = []
for i in range(len(a)):
   c = np.hstack((a[i], b))
   temp.append(c)
d = np.vstack(temp)

有没有更好更快的解决方案。


a = np.array([[1, 1], [2, 2], [3, 3]])
b = np.array([4, 4])

c = np.tile(b[np.newaxis,:], (a.shape[0],1))
d = np.concatenate((a,c), axis=1)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

在二维数组的每行末尾附加一个列表 的相关文章

随机推荐