Numpy 배열의 특정요소 추출하기

넘파이 배열을 사용할 때, 행렬 전체가 아닌 특정 성분 또는 구간 만을 사용할 때가 있습니다.

 

이에 대하여 넘파이는 특정 성분 또는 구간을 추출하는 편리한 기능을 제공하고 있습니다. 이번 실습에서 이러한 배열의 특정 성분들을 출력하는 것을 실습을 통해서 익혀봅시다.

---

배열을 인덱싱/슬라이싱 하는 함수들

• ndarray[n, m] : n 행 m 열의 원소를 추출
• ndarray[n, :] : n 행을 추출
• ndarray[:, m] : m열을 추출

---

import numpy as np

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

#1. 배열 array_1에 대해 2행 3열의 원소를 추출하세요.
element_1 = array_1[1, 2]
print("2행 3열의 원소는 ", element_1, " 입니다.")

#2. 배열 array_1에 대하여 3행을 추출하세요.
row_1 = array_1[2, :]
print("3행은 배열 ", row_1, " 입니다.")

#3. 배열 array_1에 대하여 2열을 추출하세요.
col_1 = array_1[:, 1]
print("2열은 배열 ", col_1, " 입니다.")

#4. x의 1행과 3행을 바꾼 행렬을 만들어보세요.
y = array_1[[2, 1, 0]]
print(y)

---

2행 3열의 원소는  2  입니다.
3행은 배열  [9 1 2]  입니다.
2열은 배열  [2 3 1]  입니다.
[[9 1 2]
 [5 3 2]
 [4 2 5]]