这是一种方法。
s = df.groupby('drug')['pid'].apply(set)result = df[df['pid'].isin(s['B'])]# pid drug# 0 1 A# 1 1 B# 2 1 C# 6 3 B# 7 3 C# 8 3 D说明
- 创建一个映射序列
s
作为一个单独的初始步骤，这样就无需为每个结果重新计算。 - 为了进行比较，请使用
set
O（1）复杂度查找。
这是一种方法。
s = df.groupby('drug')['pid'].apply(set)result = df[df['pid'].isin(s['B'])]# pid drug# 0 1 A# 1 1 B# 2 1 C# 6 3 B# 7 3 C# 8 3 D说明
s作为一个单独的初始步骤，这样就无需为每个结果重新计算。
setO（1）复杂度查找。
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
