Python:迭代嵌套字典时的冗余

时间:2016-01-14 09:36:05

标签: python excel dictionary tree iteration

我有一个嵌套字典,我试图循环以便写入excel文件。

这是启动和创建嵌套字典的代码

def tree(): return defaultdict(tree)
KMstruct = tree()
for system in sheet.columns[0]:
if system.value not in KMstruct:
    KMstruct[system.value]
    for row in range(1,sheet.get_highest_row()+1):
        if sheet['A'+str(row)].value == system.value and sheet['B'+str(row)].value not in KMstruct:
            KMstruct[system.value][sheet['B'+str(row)].value]
            if sheet['B'+str(row)].value == sheet['B'+str(row)].value and sheet['C'+str(row)].value not in KMstruct:
                KMstruct[system.value][sheet['B'+str(row)].value][sheet['C'+str(row)].value]
                if sheet['C'+str(row)].value == sheet['C'+str(row)].value and sheet['D'+str(row)].value not in KMstruct:
                    KMstruct[system.value][sheet['B'+str(row)].value][sheet['C'+str(row)].value][sheet['D'+str(row)].value]
                    KMstruct[system.value][sheet['B'+str(row)].value][sheet['C'+str(row)].value][sheet['D'+str(row)].value] = [sheet['E'+str(row)].value]

这是我循环的代码:

for key in KMstruct.keys():
r += 1
worksheet.write(r, col,     key)
for subkey in KMstruct[key]:
    if currsubkeyval != subkey:
        r += 1
        worksheet.write(r, col,     key)
    r +=1
    worksheet.write(r, col, key + '\\' + subkey)
    for item in KMstruct[key][subkey]:
        if curritemval != item:
            r +=1
            worksheet.write(r, col, key + '\\' + subkey)
        for subitem in KMstruct[key][subkey][item]:
            r += 1
            worksheet.write(r, col, key + '\\' + subkey + '\\' + item)
            worksheet.write(r, col + 1, subitem)
            curritemval = item
            for finalitem in KMstruct[key][subkey][item][subitem]:
                r += 1
                worksheet.write(r, col, key + '\\' + subkey + '\\' + item + '\\' + subitem)
                worksheet.write(r, col + 1, KMstruct[key][subkey][item][subitem])

因为我是一个菜鸟,请耐心等待我,我知道这不是那么美好。无论如何,我的问题是最后一个循环。我试图获取KMstruct[key][subkey][item][subitem]中的字符串值,但循环变量lastitem遍历键的字符串值的每个字符串(注意:键subitem包含字符串列表)。这意味着如果我只有一个要写入的值,则会写入与字符串中的字符一样多的次数。

例如:值:苹果将在新的excel行上写5次

我在这里做错了什么?

编辑:关于冗余的问题已经解决但现在我需要了解在将我的lastitem(即我的字符串列表)分配给子项密钥时我是否做错了。

2 个答案:

答案 0 :(得分:0)

您的直接问题是示例的最后一行应该是:

                worksheet.write(r,col+1, finalitem)
顺便说一句,如果您偶尔创建临时变量,那么您的代码将更容易阅读:

        subitemlist = KMstruct[key][subkey][item]
        for subitem in subitemlist:

答案 1 :(得分:0)

问题是在Python中str也是可迭代的,例如:

>>> for s in 'hello':
...    print(s)

h
e
l
l
o

因此,您需要避免在值为str时进行迭代,或将str包装在另一个可迭代(例如list)中,以便可以相同的方式处理。您构建代码的方式有点困难。例如,以下内容:

for key in KMstruct.keys():
    for subkey in KMstruct[key]:

...可以写成:

for key, value in KMstruct.items():
    for subkey, subvalue in value.items():

...在每个循环中为您提供,从而可以应用测试。

for key, val in KMstruct.items():
r += 1
worksheet.write(r, col,     key)

for subkey, subval in val.items():
    if currsubkeyval != subkey:
        r += 1
        worksheet.write(r, col,     key)
    r +=1
    worksheet.write(r, col, key + '\\' + subkey)

    for item, itemval in subval.items():
        if curritemval != item:
            r +=1
            worksheet.write(r, col, key + '\\' + subkey)

        for subitem, subitemval in itemval.items():
            r += 1
            worksheet.write(r, col, key + '\\' + subkey + '\\' + item)
            worksheet.write(r, col + 1, subitem)
            curritemval = item

            # I am assuming at this point subitemval is either a list or a str?
            if type(subitemval) == str:
                # If it is, wrap it as a list, so the next block works as expected
                subitemval = [subitemval] 

            for finalitem in subitemval:
                r += 1
                worksheet.write(r, col, key + '\\' + subkey + '\\' + item + '\\' + subitem)
                worksheet.write(r, col + 1, finalitem) # This should be finalitem, correct?

这里我们测试subitemval的类型,如果它是str将它包装在列表[str]中,那么下面的块按预期迭代。还有一个明显的错误,就是你没有在最后一行输出finalitem。如果没有您的示例数据,测试它是不可能的,但它应该在功能上等效。