1. 程式人生 > >TypeError: unorderable types: treeNode() < treeNode(),FP-growth

TypeError: unorderable types: treeNode() < treeNode(),FP-growth

機器學習第12章FP-growth演算法:

bigL = [v[0] for v in sorted(headerTable.items(), key=lambda p:p[1])]

 在python3中會報錯:TypeError: unorderable types: treeNode() < treeNode()

原因:首先檢視p[1]內容:

    for i in headerTable.items():
        print(i[1])
    print('********************************')

 輸出:

[3, <__main__.treeNode object at 0x000002030A0FC518>]
[3, <__main__.treeNode object at 0x000002030A0FC630>]
[4, <__main__.treeNode object at 0x000002030A0FC240>]
[3, <__main__.treeNode object at 0x000002030A0FC048>]
[5, <__main__.treeNode object at 0x000002030A0FC128>]
[3, <__main__.treeNode object at 0x000002030A0FC588>]

 預設比較第一列元素,但是當第一列元素相等時會比較第二列元素,而第二列元素是treeNode型別,無法比較treeNode型別的資料。

解決方法:

方法1、將p[1]轉換成str型別

bigL = [v[0] for v in sorted(headerTable.items(), key=lambda p:str(p[1]))]

 方法2、將p[1]改成p[1][0],明確指定比較的元素是第一列,如果相等則按照原有順序排列。