在 Python 中显示 Trie

Question

我刚开始在 Python 中尝试 Trie。我有这样的代码，但是当我得到输出时，我不知道如何将它转换回普通列表或字典。

输入：

TestDict = {0: {"ItemName" : "washing machine", "ItemStockLevel" : 15, "ItemCostPrice" : 129.21, "ItemSellingPrice" : 142.21},  

                1: {"ItemName": "french fries", "ItemStockLevel": 100, "ItemCostPrice": 21.51, "ItemSellingPrice": 41.77},  

                2: {"ItemName" : "magnum ice cream", "ItemStockLevel" : 30, "ItemCostPrice" : 57.32, "ItemSellingPrice" : 78.31}}

test = []  

for key in TestDict:  

    test.append(TestDict[key]["ItemName"])

_end = '_end_'  

def make_trie(list):  

   root = dict()  

   for word in list:  

       current_dict = root  

       for letter in word:  

           current_dict = current_dict.setdefault(letter, {})  

       current_dict[_end] = _end  

   return root  

print(make_trie(test))

输出：

{'w': {'a': {'s': {'h': {'i': {'n': {'g': {' ': {'m': {'a': {'c': {'h': {'i': {'n': {'e': {'_end_': '_end_'}}}}}}}}}}}}}}}, 'f': {'r': {'e': {'n': {'c': {'h': {' ': {'f': {'r': {'i': {'e': {'s': {'_end_': '_end_'}}}}}}}}}}}}, 'm': {'a': {'g': {'n': {'u': {'m': {' ': {'i': {'c': {'e': {' ': {'c': {'r': {'e': {'a': {'m': {'_end_': '_end_'}}}}}}}}}}}}}}}}}

想要得到：

{"washing machine", "french fries", "magnum ice cream"}

使用上面的输出。

Answer 1

试试这个TestDict = {0: {"ItemName" : "washing machine", "ItemStockLevel" : 15, "ItemCostPrice" : 129.21, "ItemSellingPrice" : 142.21},                  1: {"ItemName": "french fries", "ItemStockLevel": 100, "ItemCostPrice": 21.51, "ItemSellingPrice": 41.77},                  2: {"ItemName" : "magnum ice cream", "ItemStockLevel" : 30, "ItemCostPrice" : 57.32, "ItemSellingPrice" : 78.31}}test = []  for key in TestDict:      test.append(TestDict[key]["ItemName"])_end = '_end_'  def make_trie(list):     root = []   for word in list:         root.append(word)   return root  print(set(make_trie(test)))输出{'magnum ice cream', 'french fries', 'washing machine'}