Как расширить массив NumPy

Question

У меня есть 2 массива numpy, и я хочу объединить эти два массива вместе с помощью extend.Например:

a = [[1,2,3],[4,5,6],[7,8,9]]
b = [[0,0,0],[1,1,1]]

я хочу c = [[1,2,3],[4,5,6],[7,8,9],[0,0,0],[1,1,1]]

Кажется, я не могу использовать extend в качестве списка Python.в противном случае это вызовет ошибку AttributeError: 'numpy.ndarray' object has no attribute 'extend'.

В настоящее время я пытался преобразовать их в списки:

a_list = a.tolist()
b_list = b.tolist()
a_list.extend(b_list)
c = numpy.array(a_list)

Интересно, существует ли лучшее решение?

Answer 1

Использование -

np.concatenate((a, b), axis=0)

Или -

np.vstack((a,b))

Или -

a.append(b) # appends in-place, a will get modified directly

выход

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9],
       [0, 0, 0],
       [1, 1, 1]])