python - 将 Pandas Dataframe 与 Python 中的集合进行比较

Question

我有一个数据框，其中包含一些销售数据，如下所示：

df = pd.DataFrame({'bill_id': ['1001','1002','1006']})

我有另一组带有 bill_id 列表的集合：

{'1002', '10006', '1009'}

我试图bill_id在 Dataframe 和集合中找到共同点，并尝试了以下方法：

issues = list(l in set(df['bill_id']))

我收到一个错误TypeError: 'bool' object is not iterable，想知道我是否检查错误。期望输出是一个列表。

集合的预期输出如下：

{'1002'}

标签： pythonpandas

Answer 1

df = pd.DataFrame({'bill_id': ['1001','1002','1006']})
bill_id = ('1002', '10006', '1009')
set(df.bill_id.values).intersection(bill_id)