问：在数据框

Question

现在我有一个名为df的数据框，其中包含多个列。一列名为A.我有一个名为b的列表，其中包含A列中的部分数据。现在我想要过滤数据帧df，列A只包含列表b中的元素。 我使用了以下代码：

for i in b:
    df = df[df.A == i]

但是数据帧df变空了。 那么如何过滤数据帧呢？ THX

Answer 1

试试这个：

df = df[df.A.isin(b)]