Apriori算法Python实现23628

#!/usr/bin/python
# -*- coding:utf-8 -*-
from itertools import combinations


def load_data_set（）:
    ‘‘‘
    Returns:
    --------
    like this: [[1 3 4] [2 3 5] [1 2 3 5] [2 5]]
    ‘‘‘
    with open（“order2016-08-31“） as fp:
        itemid_list = []
        for line in fp:
            order_id order_lst = line.rstrip（‘\n‘）.split（‘\t‘）
            order_lst = eval（order_lst）
            if len（order_lst） > 1:
                order_lst = [int（itemid） for itemid in order_lst]
                itemid_list.append（order_lst）
    return itemid_list


def subtract_item_set（pre_discard_itemset candidate_set）:
    ‘‘‘
    首先去除候选集中不符合非频繁项集的那些元素，
    在当前候选集中去掉上一轮删除的项集，
    比如{2 3}是非频繁项集，那么就将删除candidate_set中的{2 3 x}这些项集

    Parameters:
    -----------