DjangoBB LoFi version

Полная версия: Некорректно сохраняются результаты обхода вложенных словарей в цикле for (поиск строки)

Начало » Python для новичков » Некорректно сохраняются результаты обхода вложенных словарей в цикле for (поиск строки)

to_night

Дек. 12, 2013 21:37:47

Добрый день. Написал следующий код для поиска вхождения строки в значении вложенного словаря. Совершается обход ключей и значений, данные сохраняются в пустой словарь. Непонятно, почему сохраняется только первое вхождение, остальное пропадает, тогда как инструкция print выводит все пары ключ: значение. Что не так?

search_string = 'Spam'
d = {1: {1: {'string': 'Spam', }, 2: {'string': 'Spam, Spam'},  3: {'string': 'Ham'}}}
dict_result = {}
 
for key, value in d.iteritems():
    for key1, value1 in d[key].iteritems():
        for key2, value2 in d[key][key1].iteritems():
            if search_string in str(value2):
                #print [key, key1, value2]
                #>>> [1, 1, 'Spam']
                #>>> [1, 2, 'Spam']
                dict_result.update({key: {key1: value2}})
 
print dict_result
>>> {1: {2: 'Spam'}}

4kpt_II

Дек. 12, 2013 22:22:02

По Вашему вопросу:

search_string = 'Spam'
d = {1: {1: {'string': 'Spam'}, 2: {'string': 'Spam, Spam'}, 3: {'string': 'Ham'}}}
dict_result = {}
#
for key, value in d.iteritems():
    for key1, value1 in value.iteritems():
        for key2, value2 in value1.iteritems():
            if search_string in str(value2):
                dict_result.setdefault(key, {}).update({key1: value2})
print dict_result

Но, если количество вложений не известно, то лучше использовать рекурсию. Писать рекурсию для Вашего примера не буду, но посмотрите повнимательнее на этот код:

search_string = 'Spam'
d = {1: {1: {'string': 'Spam'}, 2: {'string': 'Spam, Spam'}, 3: {'string': 'Ham'}}}
dict_result = {}
#
def find_string(dict_):
    for data in dict_.itervalues():
        if type(data) is dict:
            find_string(data)
        else:
            print data
#
find_string(d)

Будут вопросы - пишите.

P.S. Все там работает корректно