배열 다루기 | 조건적 출력(Numpy.extract)

 

 

 

조건적 출력(Extract Function)

numpy.extract(condition, arr)

 

입력 배열에서 지정한 조건을 만족하는 원소만 출력합니다.

▪Parameters

     ‣  condition : 조건적 불리언 배열, 조건 입력(TRUE/FALSE)으로 이루어진 1D Array

     ‣  arr : 입력 배열,

 

 

     ‣  Returns : ndarray, 조건을 만족하는 배열만 출력

 

 

예제(Example)

<Example 01>

import numpy as np

a = [1,2,3,4,5]
condition = [True, False, True, False, True]

print('Result : ', np.extract(condition, a))

 

Result :  [1 3 5]

 

<Example 02>

import numpy as np

a = [[1,2], [3,4], [5,6]]

print('1st result(F/T/F) : ')
print(np.extract([False,True,False],a))
print('2nd result(T/F/T) : ')
print(np.extract([True,False,True],a))

 

1st result(F/T/F) : 
[2]
2nd result(T/F/T) : 
[1 3]

 

<Example 03>

import numpy as np

a = [[1,2], [3,4], [5,6]]

print('1st result(F/T/F) : ')
print(np.extract([0,1,0],a))
print('2nd result(T/F/T) : ')
print(np.extract([1,0,1],a))

 

1st result(F/T/F) : 
[2]
2nd result(T/F/T) : 
[1 3]

Numpy 함수 모음

 

 

 

 

 

 

 

 

 

※ 이 글이 도움이 되었다면 "👆🏻구독"과 "🤍공감" 버튼을 클릭해주세요. 클릭 한번이 글 쓰는데 큰 힘이 됩니다.