파이썬 pandas apply 함수 질문

조회수 416회

아래의 내용을 만족하는 함수를 정의하려고 합니다.

  • 현재 user_id 컬럼은 index 입니다.

1) user_id 가 A 일때, A에 해당하는 모든 컬럼을 검색

  • user_id가 A 일때, eventdatetime, eventcategory, viewid, viewaction, funnel desc, viewid desc 컬럼만 검색
  • user_id가 A 이면서 viewid 컬럼 값이 "home", "inbox_detail", "inbox", "gig_detail", "thankyou" 인 로그 데이터의 모든 정보(컬럼)를 검색

위의 내용을 검색하는 find_log 함수를 정의하려고 합니다.

def find_log(user_id = None, columns = None, viewid = None):

    pass

를 기본으로 작성을 해야하는데, 어떻게 검색 코드를 작성해야하는 지 몰라 질문드립니다.

아래는 제가 작성해봤던 문법인데.. 이제 막 공부하기 시작해서 터무니없을 수도 있습니다 ㅠㅠ 가르침 부탁드리겠습니다.

    df_id = data.loc[user_id, :]
    df_id2 = data.loc[user_id, ['eventdatetime', 'eventcategory', 'viewid', 'viewaction', 'funnel desc', 'viewid desc']]

이런식으로 하나씩 정의해서 가져오는 건 알겠는데 하나의 def 에서 여러 조건을 적용하는 방법을 모르겠습니다..

  • (•́ ✖ •̀)
    알 수 없는 사용자

답변을 하려면 로그인이 필요합니다.

프로그래머스 커뮤니티는 개발자들을 위한 Q&A 서비스입니다. 로그인해야 답변을 작성하실 수 있습니다.

(ಠ_ಠ)
(ಠ‿ಠ)