05. [Python] NumPy 기초 : 배열 요소에 접근하는 방법

개요

NumPy에서 배열 요소에 접근하려면 Python의 리스트에서 처럼 인덱스를 이용합니다.
배열 요소에 접근하는 여러가지 방법을 알아보도록 하겠습니다.

실습환경

NumPy는 아나콘다에 포함되어 있으므로 Jupyter Notebook에서 import만 해주면 됩니다.
아나콘다 설치 및 사용방법은 아래 링크를 확인해 주세요.
https://iot-lab.tistory.com/113

파이썬 아나콘다(Anaconda) 설치하기

NumPy 배열 요소 접근 방법

1차원 배열의 인덱스를 지정하여 배열 요소값 가져오기

[In]

import numpy as np

a = np.array([0, 1, 2, 3, 4, 5])    # 2차원 배열 생성

print(a[3])

[결과]

1차원 배열의 인덱스를 지정하여 배열 요소값 수정하기

[In]

a[2] = 9

print(a)

[결과]

2차원 배열의 인덱스를 지정하여 배열 요소수 가져오기

[In]

import numpy as np

b = np.array([[0, 1, 2], [3, 4, 5]])    # 2차원 배열 생성

# 둘다 같은 결과
print(b[1,2])
print(b[1][2])

[결과]

2차원 배열의 인덱스를 지정하여 배열 요소수 수정하기

[In]

b[1,2] = 9      # b[1][2] = 9 도 동일하게 동작함

print(b)

[결과]

2차원 배열의 행, 열을 한번에 가져오기

[In]

import numpy as np

c = np.array([[0, 1, 2], [3, 4, 5]])    # 2차원 배열 생성

# 둘다 같은 결과
print(c[1, :])
print('---------------')

c[:, 1] = np.array([6,7])
print(c)

[결과]

 

다음에는 함수와 배열을 함께 사용하는 방법을 알아보도록 하겠습니다.

 

https://iot-lab.tistory.com/139

06. [Python] NumPy 기초 : 함수에서 배열 사용하는 방법