我想知道如何用对应的字典替换列表列表中的数字的正确方法。
目前我的代码是:
#the object list and max weight are supposed to be user input but i have hard coded it in for now to try and get other parts working first.
object_list = [(2, 70), (3, 85), (5, 120)] # the first number in the tuple is my weight and the second is my cost
max_weight = 17 #this is the number each list in the list of lists add to
#seperatng weights and costs into a string then making a dictionary out of it.
weight_values = [int(i[0]) for i in object_list]
cost_values = [int(i[1]) for i in object_list]
dictionary = dict(zip(weight_values, cost_values))
假设我有列表(添加到最大重量的所有可能组合):
[[3, 2, 2, 2, 2, 2, 2, 2], [3, 3, 3, 2, 2, 2, 2], [3, 3, 3, 3, 3, 2], [5, 2, 2, 2, 2, 2, 2], [5, 3, 3, 2, 2, 2], [5, 3, 3, 3, 3], [5, 5, 3, 2, 2], [5, 5, 5, 2]]
我生成的字典是:
dictionary = {2: 70, 3: 85, 5: 120}
我想要尝试做的是将所有 2 替换为 70,将所有 3 替换为 85,将所有 5 替换为 120(对于这种情况)。生成的字典基于用户输入,因此在另一种情况下,所有 2 可能需要替换为 35 而不是 70。我需要一种替换数字的方法,而无需特别说明是否有 2 替换为70.这个列表列表是使用这个答案生成的(递归树过早终止函数)
要替换列表中的值:
l = [[3, 2, 2, 2, 2, 2, 2, 2], [3, 3, 3, 2, 2, 2, 2], [3, 3, 3, 3, 3, 2], [5, 2, 2, 2, 2, 2, 2], [5, 3, 3, 2, 2, 2], [5, 3, 3, 3, 3], [5, 5, 3, 2, 2], [5, 5, 5, 2]]
dictionary = {2: 70, 3: 85, 5: 120}
new_l = [[dictionary[b] for b in i] for i in l]
输出:
[[85, 70, 70, 70, 70, 70, 70, 70], [85, 85, 85, 70, 70, 70, 70], [85, 85, 85, 85, 85, 70], [120, 70, 70, 70, 70, 70, 70], [120, 85, 85, 70, 70, 70], [120, 85, 85, 85, 85], [120, 120, 85, 70, 70], [120, 120, 120, 70]]
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句