數(shù)據(jù)挖掘之決策樹歸納算法的Python實現(xiàn)

引自百度：決策樹算法是一種逼近離散函數(shù)值的方法。它是一種典型的分類方法，首先對數(shù)據(jù)進行處理，利用歸納算法生成可讀的規(guī)則和決策樹，然后使用決策對新數(shù)據(jù)進行分析。本質(zhì)上決策樹是通過一系列規(guī)則對數(shù)據(jù)進行分類的過程
決策樹的算法原理：
(1)通過把實例從根節(jié)點開始進行排列到某個葉子節(jié)點來進行分類的。
(2)葉子節(jié)點即為實例所屬的分類的，樹上的每個節(jié)點說明了實例的屬性。
(3)樹的生成，開始的所有數(shù)據(jù)都在根節(jié)點上，然后根據(jù)你所設(shè)定的標(biāo)準(zhǔn)進行分類，用不同的測試屬性遞歸進行數(shù)據(jù)分析。
決策樹的實現(xiàn)主要思路如下：
(1)先計算整體類別的熵
(2)計算每個特征將訓(xùn)練數(shù)據(jù)集分割成的每個子集的熵，并將這個熵乘以每個子集相對于這個訓(xùn)練集的頻率，最后將這些乘積累加，就會得到一個個特征對應(yīng)的信息增益。
(3)選擇信息增益最大的作為最優(yōu)特征分割訓(xùn)練數(shù)據(jù)集
(4)遞歸上述過程
(5)遞歸結(jié)束條件：訓(xùn)練集的所有實例屬于同一類；或者所有特征已經(jīng)使用完畢。
代碼如下：
[python] view plain copy
    #!/usr/bin/python  
    #coding=utf-8  
    import operator  
    import math  
      
    #定義訓(xùn)練數(shù)據(jù)集  
    def createDataSet():  
      
        #用書上圖8.2的數(shù)據(jù)  
        dataSet = [  
                ['youth', 'no', 'no', 'no'],  
                ['youth', 'yes', 'no', 'yes'],  
                ['youth', 'yes', 'yes', 'yes'],  
                ['middle_aged', 'no', 'no', 'no'],  
                ['middle_aged', 'no', 'yes', 'no'],  
                ['senior', 'no', 'excellent', 'yes'],  
                ['senior', 'no', 'fair', 'no']  
        ]  
      
        labels = ['age', 'student', 'credit_rating']  
      
        return dataSet, labels  
      
    #實現(xiàn)熵的計算  
    def calShannonEnt(dataSet):  
      
        numEntries = len(dataSet)  
        labelCounts = {}  
      
        for featVect in dataSet:  
            currentLabel = featVect[-1]  
            if currentLabel not in labelCounts:  
                labelCounts[currentLabel] = 0  
            labelCounts[currentLabel] += 1  
      
        shannonEnt = 0.0  
      
        for key in labelCounts:  
            prob = float(labelCounts[key]) / numEntries  
            shannonEnt -= prob * math.log(prob, 2)  
      
        return  shannonEnt  
      
    #分割訓(xùn)練數(shù)據(jù)集  
    def splitDataSet(dataSet, axis, value):  
      
        retDataSet = []  
      
        for featVec in dataSet:  
            if featVec[axis] == value:  
                reducedFeatVec = featVec[:axis]  
                reducedFeatVec.extend(featVec[axis+1:])  
                retDataSet.append(reducedFeatVec)  
      
        return retDataSet  
      
    #一個確定“最好地”劃分?jǐn)?shù)據(jù)元組為個體類的分裂準(zhǔn)則的過程  
    def Attribute_selection_method(dataSet):  
      
        numFeatures = len(dataSet[0]) - 1  
        baseEntropy = calShannonEnt(dataSet)  
        bestInfoGain = 0.0  
        bestFeature = -1  
      
        for i in range(numFeatures):  
      
            featList = [example[i] for example in dataSet]  
            uniqueValue = set(featList)  
            newEntropy = 0.0  
      
            for value in uniqueValue:  
                subDataSet = splitDataSet(dataSet, i, value)  
                prob = len(subDataSet) / len(dataSet)  
                newEntropy += prob * calShannonEnt(subDataSet)  
      
            infoGain = baseEntropy - newEntropy  
      
            if infoGain > bestInfoGain:  
                bestInfoGain = infoGain  
                bestFeature = i  
      
        return bestFeature  
      
    #采用majorityvote策略，選擇當(dāng)前訓(xùn)練集中實例數(shù)最大的類  
    def majorityCnt(classList):  
      
        classCount = {}  
      
        for vote in classList:  
      
            if vote not in classCount.keys():  
                classCount[vote] = 0  
      
            classCount[vote] += 1  
      
        sortedClassCount = sorted(classCount.iteritems(), key=operator.itemgetter(1), reverse=True)  
      
        return sortedClassCount[0][0]  
      
    #創(chuàng)建決策樹  
    def Generate_decision_tree(dataSet, labels):  
        classList = [example[-1] for example in dataSet]  
      
        # 訓(xùn)練集所有實例屬于同一類  
        if classList.count(classList[0]) == len(classList):  
            return classList[0]  
      
        # 訓(xùn)練集的所有特征使用完畢，當(dāng)前無特征可用  
        if len(dataSet[0]) == 1:  
            return majorityCnt(classList)  
      
        bestFeat = Attribute_selection_method(dataSet)  
        bestFeatLabel = labels[bestFeat]  
        myTree = {bestFeatLabel: {}}  
        del(labels[bestFeat])  
        featValues = [example[bestFeat] for example in dataSet]  
        uniqueVals = set(featValues)  
      
        for value in uniqueVals:  
             subLabels = labels[:]  
             myTree[bestFeatLabel][value] = Generate_decision_tree(splitDataSet(dataSet, bestFeat, value), subLabels)  
      
        return myTree  
      
    def main():  
        print '  ____            _     _           _____              '  
        print ' |  _ \  ___  ___(_)___(_) ___  _ _|_   _| __ ___  ___ '  
        print '''''   | | | |/ _ \/ __| / __| |/ _ \| '_ \| || '__/ _ \/ _ \\'''  
        print ' | |_| |  __/ (__| \__ \ | (_) | | | | || | |  __/  __/'  
        print ' |____/ \___|\___|_|___/_|\___/|_| |_|_||_|  \___|\___|決策樹'  
        print  
        myDat, labels = createDataSet()  
        myTree = Generate_decision_tree(myDat, labels)  
        print '[*]生成的決策樹：\n',myTree  
      
    if __name__ == '__main__':  
        main() 
這里的數(shù)據(jù)也是使用書上的（《數(shù)據(jù)挖掘概念與技術(shù) 第三版》）。

運行結(jié)果：