콘솔워크

dataframe의 자료를 query문을 사용해서 필터링 하는 방법 본문

프로그래밍/python

dataframe의 자료를 query문을 사용해서 필터링 하는 방법

이휘재123 2023. 5. 12. 11:16
반응형
    def get_df_filtered(self, df_result: pd.DataFrame):
        query_string = f"(열1 > {변수1}) and (열2 > {변수2}) or (열3 > {변수3}) not (열4 > {변수4})"
        print(f"query_string: {query_string}")

        df_filtered = df_result.query(query_string)
        return df_filtered

비교 연산자, 논리 연산자와 같은 연산자를 변수로 받아와서 사용하는 것도 가능하기 때문에 df.loc[]보다 유용하게 사용할 수 있다.

반응형