[Numpy] 배열의 원소 추가/삭제

insert

insert 함수를 사용해서 배열의 원소를 추가할 수 있다.

insert 함수 인자로는 원소를 추가할 배열 변수, 원소 추가 위치, 추가할 원소, axis이다.

1차원 배열 insert

#1차원 배열 원소 추가
arr = np.arange(1,9)
arr = np.insert(arr,2,50)
print(arr)	# [ 1  2 50  3  4  5  6  7  8]

2차원 배열 insert

2차원 배열 insert는 axis 옵션을 사용하여 행 또는 열을 선택하여 insert할 수 있고, 사용하지 않으면 2차원 행렬을 1차원 행렬로 변경하여 insert 된다.

 

#2차원 배열 원소 추가
arr = np.arange(1,13).reshape(3,4)
arr = np.insert(arr,2,[51,52,53,54],axis=0)
print(arr)
'''''''

[[ 1  2  3  4]
 [ 5  6  7  8]
 [51 52 53 54]
 [ 9 10 11 12]]

'''''''

arr = np.arange(1,13).reshape(3,4)
arr = np.insert(arr,2,50,axis=1)
print(arr)
'''''''

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

'''''''

arr = np.arange(1,13).reshape(3,4)
arr = np.insert(arr,2,50)
print(arr)	# [ 1  2 50  3  4  5  6  7  8  9 10 11 12]

 

delete

delete함수를 사용해서 배열의 원소를 삭제할 수 있다.

delete함수 인자로는 원소를 삭제할 배열 변수, 원소 삭제 위치, axis이다.

 

1차원 배열 원소 삭제

arr = np.arange(1,13)
print(arr)	# [ 1  2  3  4  5  6  7  8  9 10 11 12]

arr = np.delete(arr,2)
print(arr)	# [ 1  2  4  5  6  7  8  9 10 11 12]

 

2차원 배열 원소 삭제

2차원 배열 delete는 axis 옵션을 사용하여 행 또는 열을 선택하여 delete할 수 있고, 사용하지 않으면 2차원 행렬을 1차원 행렬로 변경하여 delete된다.

arr = np.arange(1,13).reshape(3,4)
print(arr)
'''

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

'''


arr = np.delete(arr,2,axis = 1)
print(arr)
'''

[[ 1  2  4]
 [ 5  6  8]
 [ 9 10 12]]

'''

arr = np.delete(arr,2)
print(arr)	# [ 1  2  5  6  8  9 10 12]