본문으로 바로가기

numpy의 where 함수 사용법

category Software/Python 2019.08.05 08:00

머신러닝이나 딥러닝을 하다보면, numpy 모듈의 다양한 함수를 사용하게 되는 것 같습니다. 오늘은 그 중에서 은근 많이 사용되는 where 함수에 대해 이야기를 해보려고 합니다. 너무 쉽고 간단하면서 괜찮은 함수인 것 같습니다.

간단하게 numpy를 import하구요. 위 코드처럼 데이터를 numpy array로 선언해 두겠습니다.

뭐 min이나 max값을 찾거나, 그 값의 위치를 찾는 것은 argmin, argmax로 됩니다.

where함수는 조건문을 사용할때입니다. 출력은 근본적으로는 인덱스입니다. a 배열에서 1보다 적은 것은 6, 7번 데이터라는 것이죠.

그걸 슬라이싱에 사용할 수 있습니다. 그러면 1보다 작은 값을 찾아서 그 값만 반환합니다.

혹은 이렇게, 10보다 크거나 같은 값을 찾아서, 0으로 바꾸고, 아닌것은 그대로 두라는 조건문도 가능합니다. 파이썬의 for문 보다 빠르다고 알려져 있습니다.~^^

이렇게 조건을 찾아, 변경하거나, 인덱싱하는 간단한 함수가 numpy의 where함수입니다.


댓글을 달아 주세요

  1. BlogIcon 북두협객 2019.08.05 13:38 신고

    코딩의 묘미가 실행 했을 때 의도대로 버그 없이 잘 작동할 때이죠^^

  2. BlogIcon 핑구야 날자 2019.08.06 06:48 신고

    함수만 잘 사용해도 도움이 많이 되겠는데요 즐거운 하루 보내세요

  3. BlogIcon Bliss :) 2019.08.06 09:01 신고

    활용은 전혀 못하겠지만ㅎㅎㅎ 원리는 간단하고 기능은 편리한 함수이네요^^ 무더위에 건강 유의하시길요!

  4. BlogIcon 멜로요우 2019.08.07 08:48 신고

    잘보고 갑니다. 매번 핑크윙크님 블로그를 볼때마다 전문적인 느낌이 듭니다.ㅋ

  5. BlogIcon peterjun 2019.08.26 06:35 신고

    함수 사용법은 공부하다 보면 계속 찾게 되는 정보인 것 같아요.
    많은 분들께 도움이 될 것 같네요. ㅎㅎ

  6. BlogIcon 공수래공수거 2019.08.26 06:37 신고

    8월 마지막주 힘차게 시작하시기 바랍니다.^^

  7. BlogIcon anne_loving 2019.08.28 02:01 신고

    ㅇㅂㅇ 이런표정으로 봤어요
    멋있으세요!!