배열 다루기 | 조건적 출력(numpy.compress)

 

 

 

 

조건적 출력(Compress Function)

numpy.compress(condition, a, axis=None, out=None)

 

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

▪Parameters

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

     ‣  a : 입력 배열,

     ‣  axis : 축 설정(선택 사항), 계산이 진행되는 축 설정

     ‣  out : 출력 배열(선택 사항), 

 

 

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

 

 

예제(Example)

<Example 01>

import numpy as np

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

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

Result

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.compress([False,True,False],a, axis=0))
print('2nd result(T/F/T) : ')
print(np.compress([True,False,True],a, axis=0))

Result

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

<Example 03>

import numpy as np

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

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

Result

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

Numpy 함수 모음

 

 

 

 

 

 

 

 

 

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