numpy.delete¶

numpy.delete 함수는 인덱스를 이용해서 어레이의 요소 또는 서브어레이 (sub-array)를 제거합니다.

축을 지정하면 지정한 축에 대해 서브어레이가 제거된 어레이를 반환합니다.

예제1¶

import numpy as np

a = np.arange(12).reshape(3,4)

print(a)
print(np.delete(a, 5))
print(np.delete(a, 1, axis=1))

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
[ 0  1  2  3  4  6  7  8  9 10 11]
[[ 0  2  3]
 [ 4  6  7]
 [ 8 10 11]]

np.delete(a, 5)와 같이 축이 지정되지 않으면 1차원으로 변환된(flattened) 어레이에서 지정한 인덱스의 값을 제거합니다.

np.delete(a, 1, axis=1)과 같이 축을 지정하면, 축을 따라 지정한 인덱스의 서브어레이를 제거한 어레이를 반환합니다.

a = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
b = [2, 4, 6]

print(np.delete(a, b))

[0 1 3 5 7 8 9]

여러개의 요소를 동시에 제거하려면 인덱스를 어레이 형태로 입력해줍니다.