본문 바로가기
data science/python

numpy 조건에 맞는 값들만 뽑아내기

by 꼰대코더 2024. 1. 7.

b > 3 를 만족하는 인덱스를 a 에서 뽑아내기

a = np.array([1,1,1,2,2,2])
b = np.array([4,6,1,8,2,1])

 

  • 방법-1
new_a = a[ b > 3 ]

 

  • 방법-2
new_a = a[ np.where(b>3)[0] ]

 

'data science > python' 카테고리의 다른 글

오버랩핑된 문자열 잇기  (0) 2024.02.02
tuple 변경하기  (0) 2024.01.28
list ( [ , ], [ , ] ... ) 최대, 최소값 구하기  (0) 2024.01.07
2중 in 을 사용하여 리스트 만들기  (0) 2024.01.06
(python) logging  (1) 2023.12.20