Numpy - 비교(Comparison)

any & all

any은 어느 하나라도 만족하면 True 아니면 False
all은 모두가 만족하면 True 아니면 False

arange 가정

In

a = np.arange(10)
a

기본형

예시1

In

a>5

Out

array([False, False, False, False, False, False,  True,  True,  True, True])

any

예시1

In

np.any(a>5), np.any(a<0)

Out

(True, False)

all

예시

In

np.all(a>5), np.all(a<10)

Out

(False, True)

np.where

이 조건에 맞는 index 값을 반환하는 용도로 많이 쓰인다

test_a 가정

In

test_a = np.array([1, 3, 0], float)

np.where(조건, True, False)

조건에 맞으면 True에 있는 값이 출력 아니면 False에 있는 값이 출력된다.

예시1

In

np.where(test_a > 0, 3, 2) # np.where(조건, True, False)

Out

array([3, 3, 2])

np.where(a>0)

index를 반환한다.

예시1

In

np.where(test_a > 0)

Out

(array([0, 1], dtype=int64),)

argmax & argmin

최대값 또는 최소값의 index 출력

예시1 (기본형)

In

a = np.array([1,2,4,5,8,78,23,3])
np.argmax(a) , np.argmin(a)

Out

(5, 0)

예시2 (axix를 넣은 형태 즉 축을 넣은 형태)

In

a=np.array([[1,2,4,7],[9,88,6,45],[9,76,3,4]])
np.argmax(a, axis=1) , np.argmin(a, axis=0)

Out

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