numpy.insert¶

numpy.insert 함수는 주어진 축에 대해 주어진 인덱스 앞에 값을 삽입합니다.

예제1¶

import numpy as np

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

print(a)
print(b)

[1 2 3]
[1 5 2 3]

어레이 a의 두번째 위치(index=1)에 값 5를 삽입합니다.

import numpy as np

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


print(a)
print(b)

[[1 1]
 [2 2]
 [3 3]]
[1 5 1 2 2 3 3]

axis를 지정해주지 않아서 (axis=None), 우선 1차원 어레이로 변환(flatten)한 후 값을 삽입합니다.

import numpy as np

a = np.array([[1, 1], [2, 2], [3, 3]])
b = np.insert(a, 1, 5, axis=0)
c = np.insert(a, 1, 5, axis=1)

print(a)
print(b)
print(c)

[[1 1]
 [2 2]
 [3 3]]
[[1 1]
 [5 5]
 [2 2]
 [3 3]]
[[1 5 1]
 [2 5 2]
 [3 5 3]]

어레이 b는 axis=0으로 지정한 후 값을 삽입한 어레이입니다. 첫번째 축을 따라 5가 삽입되었습니다.

어레이 c는 axis=1으로 지정한 후 값을 삽입한 어레이입니다. 두번째 축을 따라 5가 삽입되었습니다.

import numpy as np

a = np.arange(6)
b = np.insert(a, (1, 3, 5), 10)

print(a)
print(b)

[0 1 2 3 4 5]
[ 0 10  1  2 10  3  4 10  5]

인덱스를 튜플의 형태((1, 3, 5))로 입력해주면,

어레이 a의 1, 3, 5 위치의 앞에 값 10을 삽입합니다.