Pandas DataFrame过滤列A，具体取决于列B是否包含x以表示A中的值组

Question

我想根据列中的值是否包含该值，df对列下面的DataFrame进行过滤。refreftype'P'In [32]: dfOut[32]:    ref type0    1    P1    1    C2    1    A3    2    C4    3    P5    3    P6    4    P7    4    A8    5    C9    5    A此处，ref值1、3和4至少包含一行，其值'P'在column type，而2和5则没有。我试图用ref2和5过滤掉任何行，以便最终输出是：In [34]: dfOut[34]:    ref type0    1    P1    1    C2    1    A4    3    P5    3    P6    4    P7    4    A我该怎么做（最好一步一步完成）？

Answer 1

使用groupby和filter：

df.groupby('ref').filter(lambda x : ('P' in x['type'].values))

返回：

   ref type

0    1    P

1    1    C

2    1    A

4    3    P

5    3    P

6    4    P

7    4    A