넘파이 조건에 맞는 값의 위치 반환

넘파이 조건에 맞는 값의 위치 반환¶

In [1]:

import numpy as np
# 예시 넘파이 어레이 생성
arr = np.array([[1, 2, 3, 4], 
                [5, 6, 7, 8]])

In [2]:

# 예시 에레이를 0차원의 
arr.shape

Out[2]:

(2, 4)

In [3]:

# 2는 (0 0번째 1차원 1번째 위치)에 있음
np.where(arr == 2)

Out[3]:

(array([0]), array([1]))

In [4]:

# 7는 (0차원 1번째 1차원 1번째 위치)에 있음
np.where(arr == 7)

Out[4]:

(array([1]), array([2]))

In [5]:

# 2보다 작거나 같은 숫자 차원 및 위치
np.where(arr <= 2)

Out[5]:

(array([0, 0]), array([0, 1]))

In [6]:

# np.where의 결과를 이용하면 조건에 맞는 값을 인덱싱 할 수 있습니다.
# 2보다 작거나 같은 숫자만 반환하기
arr[np.where(arr <= 2)]

Out[6]:

array([1, 2])

In [7]:

# 단 해당 위치에 값을 불러와 1차원 어레이로 나열 하여 결과를 도출 합니다.
arr[np.where(arr <= 7)]

Out[7]:

array([1, 2, 3, 4, 5, 6, 7])