关联分析-FP树法python注解

定义树的结构

class treeNode:
    def __init__(self,nameValue,numOccur,parentNode):
        self.name = nameValue #值
        self.count = numOccur #计数
        self.nodeLink = None #横向链
        self.parent = parentNode #父亲节点
        self.children = {}   #儿子节点
    def inc(self,numOccur):
        self.count += numOccur
    def disp(self,ind = 1):   #输出显示
        print ' ' * ind,self.name,' ',self.count
        for child in self.children.values():
            child.disp(ind + 1)

创建树

主要目的是将数据一行一行插入字典树里面,但是先进行了过滤操作,删除了一些不可能出现的项集,并且对记录进行了排序.

#dataSet是记录,minSup是最小支持度
def createTree(dataSet,minSup=1):
    #对每个元素进行计数
    headerTable = {}