파이썬 NumPy 배열, np.nditer을 사용하여 for문으로 다루기

NumPy 배열 반복하기

1차원(1D) NumPy 배열을 다루는 경우, 모든 요소를 반복(iterate)하는 것은 아주 간단합니다!

for x in my_array:
    ...

하지만 2차원(2D) NumPy 배열을 다루는 경우는 좀 더 복잡합니다.

2차원 배열은 여러 개의 1차원 배열로 구성되어 있기 때문입니다.

다차원 배열의 모든 개별 요소를 명시적으로 반복하려면, 다음과 같은 문법을 사용해야 합니다:

for x in np.nditer(my_array):
    ...

참고: Python 세션에는 아래와 같은 두 개의 NumPy 배열이 이미 제공되어 있다고 가정합니다.

• np_height: 메이저리그 야구 선수들의 키(height)를 담고 있는 1차원 NumPy 배열
• np_baseball: 선수들의 키(첫 번째 열)와 몸무게(두 번째 열)를 포함하는 2차원 NumPy 배열