[파이썬 강의] 67. 배열 요소 삭제하기

배열에 요소를 추가할 때는 np.delete 라는 함수를 사용합니다. 사용 방법은 아래와 같습니다. 

np.delete(배열, 인덱스, 축) 

1차원 배열을 예로 들어봅시다. 원소가 1,2,3,4인 1차원 배열을 하나 정의하겠습니다. 

>>> import numpy as np
>>> v1=np.array([1,2,3,4])

두번째 원소를 삭제하고 싶은 경우 index에 1을 입력하면 됩니다. 

>>> np.delete(m1,1,axis=0)
array([1, 3, 4])

2차원 배열을 예로 들어봅시다. 2차원 배열을 하나 정의하겠습니다. 

>>> m1=np.array([[1,2],[3,4]])

[3,4] 를 제거해봅시다. 

>>> np.delete(m1,1,axis=0)
array([[1, 2]])

2차원 배열을 다시 정의합시다. 

>>> m1=np.array([[1,2],[3,4]])

이번에는 1을 제거하려고 시도해보겠습니다. 

>> >m1=np.array([[1,2],[3,4]])
>>> np.delete(m1,0,axis=1)
array([[2],
       [4]])

axis1은 1->2 와 3->4 두개이므로, 각각의 0번 인덱스 원소가 제거됩니다.