판다스 데이터프레임 중에서 특정한 값을 갖는 위치의 열이름 추출
조회수 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
댓글 입력