更多pythonic方式来改变多级字典值

时间:2015-11-09 10:18:29

标签: python dictionary nested

例如,我有以下词典:

{'foo': 'test', 
 'bar': {'test1': 'some text here'},
 'baz': {'test2': {'test3': 'some text here', 'test4': 'some text here'}}}

或类似的东西,可能更多级别。

现在问题是,我想将'some text here'更改为'A test'。是的,我可以使用很多for循环,如下面的代码:

d = {'foo': 'test',
     'bar': {'test1': 'some text here'},
     'baz': {'test2': {'test3': 'some text here',
                       'test4': 'some text here'}}}

for i in d:
    if d[i] == 'some text here':
        d[i] = 'A test'

    elif type(d[i]) == dict:
        for j in d[i]:
            if d[i][j] == 'some text here':
                d[i][j] = 'A test'

            elif type(d[i][j]) == dict:
                for n in d[i][j]:
                    if d[i][j][n] == 'some text here':
                        d[i][j][n] = 'A test'

__import__('pprint').pprint(d)

输出:

{'bar': {'test1': 'A test'},                                                    
 'baz': {'test2': {'test3': 'A test', 'test4': 'A test'}},
 'foo': 'test'}

但是我认为这不是一个好方法......有什么想法吗?

4 个答案:

答案 0 :(得分:3)

这看起来像递归的好例子。

import re

def replace_rec(data, search, replace, _pattern=None):
    if _pattern is None:
        _pattern = re.compile(r'^%s$' % search)
    for k, v in data.items():
        try:
            data[k] = _pattern.sub(replace, v)
        except TypeError:
            try:
                replace_rec(data[k], search, replace, _pattern=_pattern)
            except AttributeError:
                # Leave any other types as they are.
                continue

将它用于您的示例:

>>> data = {
...     'foo': 'test',
...     'bar': {'test1': 'some text here'},
...     'baz': {'test2': {'test3': 'some text here', 'test4': 'some text here'}},
...     'loc': [1, 2, 3],
...     'fp': 'foo some text here foo',
... }
>>> replace_rec(data, 'some text here', 'A test')
>>> pprint.pprint(data)
{'bar': {'test1': 'A test'},
 'baz': {'test2': {'test3': 'A test', 'test4': 'A test'}},
 'foo': 'test',
 'fp': 'foo some text here foo',
 'loc': [1, 2, 3]}

答案 1 :(得分:1)

略有替代版本。这可以正确处理词典中的非字符串,并且只替换完全匹配的文本。

def replace(d, find_text, replace_text):
    for k, v in d.items():
        if isinstance(v, dict):
            replace(v, find_text, replace_text)
        elif isinstance(v, str):
            if v == find_text:
                d[k] = replace_text

d = {
    'test': 'dont change some text here',
    'ignore' : 42,

    'foo': 'test', 
    'bar': {'test1': 'some text here'},
    'baz': {'test2': {'test3': 'some text here', 'test4': 'some text here'}}}       

replace(d, 'some text here', 'A test')

__import__('pprint').pprint(d)

这会显示:

{'bar': {'test1': 'A test'},
 'baz': {'test2': {'test3': 'A test', 'test4': 'A test'}},
 'foo': 'test',
 'ignore': 42,
 'test': 'dont change some text here'}

答案 2 :(得分:1)

递归答案都很有趣和游戏,但你可以通过注意字典是可变的来获得更多的Pythonic:

首先获取所有词典的引用(在所有级别),然后更改它们。

npm install

答案 3 :(得分:0)

只要你改变的价值不是太多毛,这里就是一个单行:

In [29]: data = dict(foo=12, bar=dict(dd=12), ex=dict(a=dict(b=dict(v=12))))

In [30]: json.loads(re.sub("12", "33", json.dumps(data)))
Out[30]: {'bar': {'dd': 33}, 'ex': {'a': {'b': {'v': 33}}}, 'foo': 33}

编辑,每凯文,简单的替换甚至不需要re

json.loads(json.dumps(data).replace("12", "33"))