선택한 열과 조건부 선택으로 새 DF 작성

Question

15 개의 열이있는 DF가 있습니다.이 중 5 개의 유용한 열만 있습니다. 또한 조건을 만족시키는 행만 원합니다. 이것들은 jupyter 노트북에 사용되는 코드입니다.

cancelled = df[df['CANCELLED'] == 1] 

cols_of_interest = ['DAY_OF_WEEK','AIRLINE','ORIGIN_AIRPORT','CANCELLED','DEPARTURE_DELAY'] 

new_df = cancelled[cols_of_interest] 

지금 내가 가지고 내 선택된 열 new_df하고는 = 1. '취소'그러나이 일을 더 파이썬 방법이 조건을 만족?

내가 얻을 수있는 의견을 감사하십시오.

Answer 1

당신은 그것에 대해 .loc를 사용할 수 있습니다

cols_of_interest = ['DAY_OF_WEEK','AIRLINE','ORIGIN_AIRPORT','CANCELLED','DEPARTURE_DELAY'] 

new_df = df.loc[df['CANCELLED'] == 1, cols_of_interest] 

더 자세한 정보를 the doc에.