python - 轴错误中不包含标签 - pandas dataframe

Question

我有一个名为 df 的数据框，看起来像 -

pageno     entity          code         rawentity 
17727425   SAUDI           CBCNTRY      saudi 
17727425   GARRA           DRWRNAME     garra
17727425   PO BOX          RBCNTRY      po box 
17727425   NEW ZEALAND     DRWRCNTRY    new zealand

我还有一个包含国家/地区名称的国家/地区列表。它的类型为“列表”。

我试图仅保留那些代码为 - CBCNTRY 或 RBCNTRY 或 DRWRCNTRY 的值，并且实体应位于国家/地区列表。

我编写的代码是-

        for row in df.itertuples():

            if(row.code in ['DRWRCNTRY', 'RBCNTRY', 'CBCNTRY']):
                if(row.entity not in countrylist):
                    df.drop((row.index), inplace=True)

但是我收到以下错误 -

Error is: labels [<built-in method index of Pandas object at 0x0000020A1BCE4EB8>] not contained in axis

我只是想知道为什么我的方法是错误的，除了这个方法之外我还能做些什么更好的事情。

我已经搜索过这个错误，但无法得到满意的答案。

Answer 1

按照 Gerardo 的建议，使用 pd.isin 并使用 bool 运算符组合表达式:

countrylist = ['SAUDI']
codelist = ['DRWRCNTRY', 'RBCNTRY', 'CBCNTRY']
df = df[(df['code'].isin(codelist)) & (df['entity'].isin(countrylist))]

结果

     pageno entity     code rawentity
0  17727425  SAUDI  CBCNTRY     saudi