1. 단일 요소 선택:

   import numpy as np
   
   arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
   selected_element = arr[1, 2]  # 2차원 배열에서 (1, 2) 위치의 요소 선택
   print(selected_element)  # 출력: 6
   

2. 행 또는 열 선택:

   import numpy as np
   
   arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
   selected_row = arr[1]  # 인덱스 1의 행 선택
   selected_column = arr[:, 2]  # 인덱스 2의 열 선택
   print(selected_row)  # 출력: [4 5 6]
   print(selected_column)  # 출력: [3 6 9]
   

3. 조건에 맞는 요소 선택:

   import numpy as np
   
   arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
   selected_elements = arr[arr > 5]  # 조건 arr > 5을 만족하는 요소 선택
   print(selected_elements)  # 출력: [6 7 8 9]
   

4. 다중 인덱싱:

   import numpy as np
   
   arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
   selected_elements = arr[[0, 2], [1, 2]]  # (0, 1)과 (2, 2) 위치의 요소 선택
   print(selected_elements)  # 출력: [2 9]
   

NumPy의 배열 인덱싱 기법은 다차원 배열에서 원하는 요소를 선택하는 다양한 방법을 제공합니다. 이를 통해 배열의 특정 요소를 선택하거나 특정 조건을 만족하는 요소를 선택할 수 있습니다.

참고 문서:

• NumPy 배열 인덱싱