Вставка строки в определенном месте в массиве 2d в numpy?

У меня есть 2d-массив в numpy, где я хочу вставить новую строку. Следующий вопрос Numpy - добавление строки в массив может помочь. Мы можем использовать numpy.vstack, но он складывается в начале или в конце. Кто-нибудь может помочь в этом отношении.

Ответы

Ответ 1

Возможно, вы ищете numpy.insert

>>> import numpy as np
>>> a = np.zeros((2, 2))
>>> a
array([[ 0.,  0.],
       [ 0.,  0.]])
# In the following line 1 is the index before which to insert, 0 is the axis.
>>> np.insert(a, 1, np.array((1, 1)), 0)  
array([[ 0.,  0.],
       [ 1.,  1.],
       [ 0.,  0.]])
>>> np.insert(a, 1, np.array((1, 1)), 1)
array([[ 0.,  1.,  0.],
       [ 0.,  1.,  0.]])