定义树的结构
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 = {}