如何在遍历字典时在字典中添加值？

Question

我有一本字典，根据我拥有的值，我想创建另一个“变量”；并将其放入字典中。我能够创建另一个变量，但我试图将它放入字典中的方法不起作用。我的代码是：

mydict = {'1':('a','example'),'2':('b','example'),'3':('c','example'),'4':('d','example')}

def to_use (dictionary):
    for key, val in dictionary.iteritems():
        if len(val) == 1:
            if val[0] == 'a':
                new = 'var_a'
            elif val[0] == 'b':
                new = 'test_b'
            elif val[0] == 'c':
                new = 'var_c'
            elif val[0] == 'd':
                new = 'test_d'
        new_dict[key.lower()].append(new)

to_use(mydict)

函数中的最后一行应该创建另一个字典（但不是），因为我不确定是否可以在不创建新字典的情况下访问我的最终字典。 我想要的是：

mydict = {'1':('a','example','var_a'),'2':('b','example','test_b'),'3':('c','example','var_c'),'4':('d','example','test_d')}

如果有人有任何解决方案.. 谢谢！

Answer 1

一种方法是为每个元组创建一个要添加的值的字典，这将简化例如a 和 var_a，并用它来创建一个像这样的新字典

mydict = {'1':('a','example'),'2':('b','example'),'3':('c','example'),'4':('d','example')}

def to_use(dictionary):

    # Mapping of element to add to tuple
    mapping = {'a': 'var_a', 'b': 'test_b', 'c': 'var_c', 'd': 'test_d'}

    #Iterate through dictionary
    for key, val in dictionary.iteritems():
        #For each key, add a 1 element-tuple to each value
        dictionary[key] = val + (mapping[val[0]],)

    return dictionary

print(to_use(mydict))

输出将是

{'1': ('a', 'example', 'var_a'), 
'2': ('b', 'example', 'test_b'), 
'3': ('c', 'example', 'var_c'), 
'4': ('d', 'example', 'test_d')}

Answer 2

我认为代码存在多个问题。 让我试着一一解决。

1) 元组是不可变的 您的字典值是元组，因此无法更改。 您可以或者将它们更改为['a','example'] 而非('a','example') 的列表。

或在循环期间用新元组替换它们。

dictionary[key] = val + (new,)

2) 您正在使用 if len(val) == 1: 检查长度是否为 1。对于您的示例数据，这绝不是真的。因此，当您到达new_dict[key.lower()].append(new) 时，没有new 变量。

我建议删除 if 语句。

3) 您从未将 new_dict 声明为字典，因此 new_dict[key.lower()].append(new) 失败。

使用元组尝试以下代码：

mydict = {'1':('a','example'),'2':('b','example'),'3':('c','example'),'4':('d','example')}

def to_use (dictionary):
    for key, val in dictionary.iteritems():
        if val[0] == 'a':
            new = 'var_a'
        elif val[0] == 'b':
            new = 'test_b'
        elif val[0] == 'c':
            new = 'var_c'
        elif val[0] == 'd':
            new = 'test_d'
        else:
            new = None
        if new is not None:
            dictionary[key] = val + (new,)

to_use(mydict)

或使用列表的以下代码：

mydict = {'1':['a','example'],'2':['b','example'],'3':['c','example'],'4':['d','example']}

def to_use (dictionary):
    for key, val in dictionary.iteritems():
        if val[0] == 'a':
            new = 'var_a'
        elif val[0] == 'b':
            new = 'test_b'
        elif val[0] == 'c':
            new = 'var_c'
        elif val[0] == 'd':
            new = 'test_d'
        else:
            new = None
        if new is not None:
            dictionary[key].append(new)

to_use(mydict)

两者都应该按您的预期工作。

Answer 3

我通过将我创建的新字典声明为 defaultdict(list) 找到了解决方案。

感谢您的帮助！