apriori算法 python代码

当然可以，下面是一个简单的apriori算法的Python代码：

def apriori(transactions, support_threshold):
    # 获取所有项集中的唯一项
    unique_items = set(item for transaction in transactions for item in transaction)

    # 初始化频繁1项集
    frequent_items = {}
    for item in unique_items:
        frequent_items[frozenset([item])] = sum(1 for transaction in transactions if item in transaction)

    # 生成频繁项集
    while frequent_items:
        new_frequent_items = {}
        for itemset in frequent_items.keys():
            for item in unique_items:
                if item not in itemset:
                    new_itemset = itemset.union([item])
                    count = sum(1 for transaction in transactions if new_itemset.issubset(transaction))
                    if count >= support_threshold:
                        new_frequent_items[new_itemset] = count
        frequent_items = new_frequent_items
        yield frequent_items

希望对你有所帮助！