배열에 원소 추가 및 삭제

• insert 함수

  몇번째 자리에 어떤 원소를 넣을 것인지를 할 수 있게 하는 함수

1차원 함수

arr = [1, 2, 3, 4, 5, 6, 7, 8]
arr.insert(2, 50)
print(arr)
[1, 2, 50, 3, 4, 5, 6, 7, 8]

arr = np.arange(1,9)
arr = np.insert(arr, 2, 50)
print(arr)
[1, 2, 50, 3, 4, 5, 6, 7, 8]

2차원 함수 - axis 값 기준으로 원소 삽입 가능

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

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]]

• delete 함수

  axis 값을 정하지 않으면 1차원 배열로 변경해 삭제한다.

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=0)
print(arr)
[[1 2 3 4]
 [5 6 7 8]]

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  4  5  6  7  8  9 10 11 12]

배열 간의 병합

• append 함수

  배열의 마지막 부분에 추가해 병합하는 함수

  axis = 0 : x축 기준으로 병합

  axis = 1 : y축 기준으로 병합

  axis 없으면 1차원 배열로 변형해서 병합한다.

arr1 = np.arange(1, 13).reshape(3,4)
arr2 = np.arange(13, 25).reshape(3,4)

arr3 = np.append(arr1, arr2, axis = 0)
print(arr3)
[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]
 [13 14 15 16]
 [17 18 19 20]
 [21 22 23 24]]

arr3 = np.append(arr1, arr2, axis = 1)
print(arr3)
[[ 1  2  3  4 13 14 15 16]
 [ 5  6  7  8 17 18 19 20]
 [ 9 10 11 12 21 22 23 24]]

arr3 = np.append(arr1, arr2)
print(arr3)
[ 1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24]

• vstack : axis = 0 기준으로 병합하는 함수
• hstack : axis = 1 기준으로 병합하는 함수

arr1 = np.arange(1, 7).reshape(2, 3)
arr2 = np.arange(7, 13).reshape(2, 3)

arr3 = np.vstack((arr1, arr2))
print(arr3)
[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]

arr3 = np.hstack((arr1, arr2))
print(arr3)
[[ 1  2  3  7  8  9]
 [ 4  5  6 10 11 12]]

• concatenate 함수 - 중괄호 이용

arr1 = np.arange(1, 7).reshape(2, 3)
arr2 = np.arange(7, 13).reshape(2, 3)

arr3 = np.concatenate([arr1, arr2], axis = 0)
print(arr3)
[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]

arr3 = np.concatenate([arr1, arr2], axis = 1)
print(arr3)
[[ 1  2  3  7  8  9]
 [ 4  5  6 10 11 12]]