[Python] Numpy 항목 삭제하기 delete

numpy에선 delete()를 사용해 항목을 삭제할 수 있다.

>>> x
array([17,  7, 19, 17,  6,  4, 15])

# 3번 인덱스를 삭제하기
>>> x = np.delete(x, 3)
>>> x
array([17,  7, 19,  6,  4, 15])

# 1, 5번 인덱스 삭제하기
>>> np.delete(x, [1, 5])	#삭제할 인덱스가 2개 이상이면, 리스트에 담아서 처리
array([17, 19,  6,  4])

 

다차원은 축을 기준으로 삭제가 가능

>>> X
array([[36, 66, 62, 17, 50],
       [58, 47,  6, 45, 72],
       [11,  1,  2, 85, 97],
       [12, 64, 21,  3, 33]])

>>> np.delete(X, 1)
array([36, 62, 17, 50, 58, 47,  6, 45, 72, 11,  1,  2, 85, 97, 12, 64, 21,
        3, 33])


>>> np.delete(X, 2 , axis = 0)		#2번 행을 삭제
array([[36, 66, 62, 17, 50],
       [58, 47,  6, 45, 72],
       [12, 64, 21,  3, 33]])

>>> np.delete(X, 1, axis = 1)		#1번 열을 삭제
array([[36, 62, 17, 50],
       [58,  6, 45, 72],
       [11,  2, 85, 97],
       [12, 21,  3, 33]])

>>> np.delete(X, [0,2,3], axis = 1) # 0,2,3번 열을 해제
array([[66, 50],					# 삭제할 열이나 행이 
       [47, 72],					# 여러개일 때는 리스트에 담아서 처리한다.
       [ 1, 97],
       [64, 33]])