добавить список в конце каждой строки 2D массива

Question

Я хочу, чтобы добавить в список/1d массив (б) в конце каждой строки в 2d массив (а)

вход:

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

мой код:

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

есть ли лучшее и быстрое решение для этого.

Answer 1

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)