根据现有列中的2个条件创建新的熊猫数据框

Question

这里的问题很简单：我正在基于合同报告创建一个新的数据框，我只需要2个状态。我有第一个条件，但是在添加第二个条件时遇到了麻烦。

df2 = df[(df['[PCW] Contract Status'] == "Draft")]

其他状态为“草稿修订”。所以我基本上希望它读起来像

df2 = df[(df['[PCW] Contract Status'] == "Draft", "Draft Amendment")]

Answer 1

您可以使用isin()。

df2 = df[df['[PCW] Contract Status'].isin(["Draft", "Draft Amendment"])]

否则，您可以更早地创建所需变量的列表，然后在isin()中添加列表的名称。