본문 바로가기
머신러닝을 위한 파이썬 기초/넘파이

[파이썬 넘파이] 배열에서 True 인 원소의 인덱스를 추출하기

by bigpicture 2023. 9. 18.
반응형

True 와 False를 원소로 갖는 배열을 하나 만들어 줍니다. 

a=np.array([True,True,False,False,True])

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


True 인 행의 index만 추출하는 방법은 아래와 같습니다. 

>>> a.nonzero()
(array([0, 1, 4], dtype=int64),)

 

반응형

댓글