给出一个Apriori算法的示例,基于mlxtend和python
时间: 2023-05-22 14:03:40 浏览: 171
好的,下面是一个基于mlxtend和python的Apriori算法的示例:
```python
from mlxtend.frequent_patterns import apriori
from mlxtend.preprocessing import TransactionEncoder
import pandas as pd
# 构造样本数据
dataset = [['Apple', 'Beer', 'Rice', 'Chicken'],
['Apple', 'Beer', 'Rice'],
['Apple', 'Beer'],
['Apple', 'Bananas'],
['Milk', 'Beer', 'Rice', 'Chicken'],
['Milk', 'Beer', 'Rice'],
['Milk', 'Beer'],
['Apple', 'Bananas']]
# 转换为二维矩阵
te = TransactionEncoder()
te_ary = te.fit(dataset).transform(dataset)
df = pd.DataFrame(te_ary, columns=te.columns_)
# 计算频繁项集
frequent_items = apriori(df, min_support=0.5, use_colnames=True)
# 显示频繁项集
print(frequent_items)
```
该示例代码中的数据集为一个购物清单,包含了8个交易记录和5种物品。我们使用mlxtend中的`TransactionEncoder`对数据进行转换,然后计算支持度不低于0.5的频繁项集,最终输出结果。
希望这个示例对您有所帮助。
阅读全文