python - 与广播连接

Question

考虑以下数组:

a = np.array([0,1])[:,None]
b = np.array([1,2,3])

print(a)
array([[0],
       [1]])

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

是否有一种简单的方法可以以广播后者的方式连接这两个数组，以获得以下结果？

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

我看到有this已关闭问题与相关问题。提出了一种涉及 np.broadcast_arrays 的替代方案，但是我无法设法使其适应我的示例。除了 np.tile/np.concatenate 解决方案之外，还有其他方法可以做到这一点吗？

Answer 1

您可以通过以下方式进行

import numpy as np
a = np.array([0,1])[:,None]
b = np.array([1,2,3])
b_new = np.broadcast_to(b,(a.shape[0],b.shape[0]))
c = np.concatenate((a,b_new),axis=1)
print(c)