판다스 데이터프레임 중에서 특정한 값을 갖는 위치의 열이름 추출

조회수 1730회

판다스 데이터 프레임에서 특정 값을 알고 있는 경우 그 값의 열이름을 어떻게 출력할까요?

그리고 특정값을 알고 있는 경우 그 값의 위치 출력을 어떻게 해야하는지 궁금합니다!

계속 검색해도 안나와서 질문 남깁니다. 감사합니다.

1 답변

  • >>> df = pd.DataFrame({"a":[1,2,3,1], "b":[3,3,1,1]})
    >>> df
       a  b
    0  1  3
    1  2  3
    2  3  1
    3  1  1
    >>> df==1
           a      b
    0   True  False
    1  False  False
    2  False   True
    3   True   True
    >>> (df==1).any(axis=0)
    a    True
    b    True
    dtype: bool
    >>> df.loc[:,(df==1).any(axis=0)]
       a  b
    0  1  3
    1  2  3
    2  3  1
    3  1  1
    >>> df.loc[(df==1).any(axis=1), :]
       a  b
    0  1  3
    2  3  1
    3  1  1
    >>> colnames = df.columns
    >>> for idx, row in df.iterrows():
        for col in colnames:
            if row[col] == 1:
                print(idx, col)
    
    
    0 a
    2 b
    3 a
    3 b
    >>> for idx, row in (df==1).iterrows():
        for col in colnames:
            if row[col]:
                print(idx, col)
    
    
    0 a
    2 b
    3 a
    3 b
    

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

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

(ಠ_ಠ)
(ಠ‿ಠ)