使用选定的列和条件选择创建新的 DF

Question

我有一个包含 15 列的 DF，我将仅参考其中的 5 个有用的列。此外，我只想要满足条件的行。这些是我的代码，在jupyter notebook中使用。

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 和选定的列并且满足 'cancelled' = 1 的条件。但是有没有更 pythonic 的方法来做到这一点？

感谢我能得到的任何意见。

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 中有更多详细信息。