python - 如何根据某些列值创建新的数据框？

Question

我有一个数据框，其中有一列 Flag，其值为 True 或 False。我想创建一个新的数据框，其列 Flag 的值必须全部为 True。

df = {'Name':['Tom', 'nick', 'krish', 'jack'],
      'Age':['12', '23', '25', '16'],
      'Flag':[True, False, False, True]}

我想要一个新的 df，其 Flags 为 True，并删除此 Flag 列:

df = {'Name':['Tom', 'jack'],
       'Age':['12', '16']}

Answer 1

使用boolean indexing与 boolean 列，因此不需要通过 True 进行比较:

df = pd.DataFrame(df)    

您只能通过列表使用DataFrame.loc来选择某些列:

df1 = df.loc[df['Flag'], ['Name','Age']]

或者使用并删除Flag使用DataFrame.pop :

df1 = df[df.pop('Flag')]

或者选择添加DataFrame.drop后删除Flag :

df1 = df[df['Flag']].drop('Flag', axis=1)

---

print (df1)
   Name Age
0   Tom  12
3  jack  16