Numpy - 행, 열, 요소 값 삭제하기

numpy에서는 np.delete()를 사용하여 배열의 행, 열, 값을 삭제할 수 있다.

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

설명을 위하여 먼저 다음과 같이 5x4 배열을 만들어 놓았다.

import numpy as np

arr = np.array( [ [5,10,15,20] ,
                 [25,30,35,40], 
                 [45,50,55,60], 
                 [65,70,75,80], 
                 [85,90,95,100] ])

---

(1) 배열에서 행 삭제하기 (axis = 0)

# 1번째 행 값인 [25,30,35,40]을 삭제하기 
arr = np.delete(arr, 1 , axis = 0)
arr

---

(2) 배열에서 열 삭제하기 (axis = 1)

# 2번째 열 값인 [15],[35],[55],[75],[95] 을 삭제하기 
arr = np.delete(arr, 2 , axis = 1)
arr

---

(3) 배열에서 특정값 하나만 삭제하기

# 배열의 13번째 값인 [70]을 삭제하기
arr = np.delete(arr, 13) # axis 값을 입력하지 않아도 된다.
arr

---

(4) 배열에서 특정값 여러개 선택하여 삭제하기

# 배열의 3번째, 7번째, 17번째 값인 [20], [40], [90] 삭제하기
arr = np.delete(arr, (3, 7, 17))
arr

배열에서 특정값만을 삭제할 때는 axis 값을 넣지 않으며, 값이 삭제가 되면 배열이 무너져서 출력되는 것을 확인할 수 있다.