FP-Growth及关联规则python代码

class treeNode:
    def __init__（self nameValue numOccur parentNode）:
        self.name = nameValue
        self.count = numOccur
        self.nodelink = None
        self.parent = parentNode      #needs to be updated
        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）

def createTree（dataSet minSup=1）: #create FP-tree from dataset but don‘t mine
    headerTable = {}
    #go over dataSet twice
    for trans in dataSet:#first pass counts frequency of occurance
        for item in trans:
            headerTable[item] = headerTable.get（item 0） + dataSet[trans]
    f