[파이썬 numpy] 배열 요소 삭제하기

[파이썬 numpy] 

배열 요소 삭제하기 

배열 요소를 삭제할 때는 np.delete 메소드를 사용합니다. 아아래와 같은 형식으로 사용합니다. 

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

1차원 배열에서 요소 삭제

1차원배열을 정의하고, 첫번째 요소를 삭제해봅시다. 

A=np.array([1,2,3,4,5])

>>> A

array([1, 2, 3, 4, 5])

>>> np.delete(A, 0)

Out[450]: array([2, 3, 4, 5])

2차원 배열에서 행 삭제

2차원배열을 정의하고 1행을 삭제해봅시다. 2차원배열에서 행방향으로 첫번째 행을 제거하는 것이므로 axis 에 0을 넣어줍니다. 

A2=np.array([[1,2,3],[4,5,6]])

>>> A2

array([[1, 2, 3],

       [4, 5, 6]])

>>> np.delete(A2,0,0)

array([[4, 5, 6]])

2차원 배열에서 열 삭제

2차원배열을 정의하고 1열을 삭제해봅시다. 2차원배열에서 열 방향으로 첫번째 열을 제거하는 것이므로 axis 에 1을 넣어줍니다. 

>>> np.delete(A2,0,1)

array([[2, 3],

       [5, 6]])