프로그래밍/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[]보다 유용하게 사용할 수 있다.
반응형