使用选定列和条件选择创建新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包含所选列，并且满足＆＃39;取消＆＃39; = 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中的更详细信息。