Python-带有字典输入的贪婪背包

Question

给定以下数据集，我正在尝试在python中实现贪婪背包算法。 输出应该是遵守限制设置的列表列表。 在下面的数据集示例中，输出应为：

out = [[C, B, D, A], [Z, F, E]]

码：

data = {
        'A': 5,
        'B': 10,
        'C': 11,
        'D': 7,
        'E': 2,
        'Z': 4,
        'F': 3
}

def greedy_algo(mystuff, limit=30):

    # Copy the dictionary to work on duplicate
    copy_stuff = dict(mystuff)
    # Initialize an output list
    outlist = []

    def keywithmaxval(d):
     """ a) create a list of the dict's keys and values; 
     b) return the key with the max value"""  
        v=list(d.values())
        k=list(d.keys())
        return k[v.index(max(v))]


    def greedy_grab(mydict):
        result = []
        total = 0
        while total <= limit:
            maxkey=keywithmaxval(mydict)
            result.append(maxkey)
            total += mydict[maxkey]
            del mydict[maxkey]
        return result

    def update_dict(mydict, mylist):
        for i in mylist:
            del mydict[i]
        return mydict     

    while len(copy_stuff) > 0:
        outlist.append(greedy_grab(copy_stuff)


    return outlist


print (greedy_algo(data, limit=30))

我遇到了那个max函数的问题，这是我的输出：

['C']
['C', 'B']
['C', 'B', 'D']
['C', 'B', 'D', 'A']
['Z']
['Z', 'F']
['Z', 'F', 'E']
Traceback (most recent call last):
  File "gre.py", line 72, in <module>
    greedy_algo(data, limit=30)
  File "gre.py", line 63, in greedy_algo
    outlist.append(greedy_grab(copy_stuff))
  File "gre.py", line 40, in greedy_grab
    maxkey=keywithmaxval(mydict)
  File "gre.py", line 28, in keywithmaxval
    return k[v.index(max(v))]
 ValueError: max() arg is an empty sequence

我猜它将一个空字符串放入max，但是我不明白为什么，while应该在使用完最后一个元素之后终止循环。 有人能帮我吗？

Answer 1

好吧，greedy_grab的第一次执行或多或少都很好（结果大于限制，因为您在插入项目后检查了限制，但不会引发任何异常）。

但是当它完成时，循环

while len(copy_stuff) > 0:
    outlist.append(greedy_grab(copy_stuff))

再次执行该函数，但是这次“ copy_stuff”字典仅包含F，E和Z。然后循环

while total <= limit:
        maxkey=keywithmaxval(mydict)
        result.append(maxkey)
        total += mydict[maxkey]
        del mydict[maxkey]

在总数达到限制之前删除mydict中的所有元素，因此最终以空dict调用keywithmaxval。 这引发了异常。

一种可能的解决方法是将“ not empty”检查添加到循环中。

while total <= limit and len(mydict) > 0:
        maxkey=keywithmaxval(mydict)
        result.append(maxkey)
        total += mydict[maxkey]
        del mydict[maxkey]

顺便说说。 PDB是您的朋友。