Python - Numpy 조건문

np.where(조건, True, False) : 단일 조건문

5 미만이면 arr을 출력, 아니면 *10을해라

arr = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

np.where(arr < 5, arr, arr * 10)
----------------------------------------------
#result
array([ 0,  1,  2,  3,  4, 50, 60, 70, 80, 90])

 

np.select : 다중 조건문 

2 미만일때는 100을 더해라
5 초과일때는 곱하기 2를 해줘라

 

조건식과 대응되는 결과 리스트를 만들어야 함 

arr = np.array([100, 101, 2, 3, 4, 5, 12, 14, 16, 18])

cond_list = [arr > 5, arr < 2]
choice_list = [arr * 2, arr +100]

np.select(cone_list, choice_list, default = arr)
------------------------------------------------------
# result
array([200, 202,   2,   3,   4,   5,  24,  28,  32,  36])