DefaultDict在两种情况下的行为都不同

时间:2017-12-12 13:32:09

标签: python python-3.x defaultdict

我有以下内容:

a = [{ "_id" : { "reportId" : "5a27cda63fff647c33a14b31" }, "amount" : 3000 },
     { "_id" : { "reportId" : "5a27cda63fff647c33a14b31", "name" : "sriram sathyan" }, "amount" : 0 },
     { "_id" : { "reportId" : "5a27cf173f978655f2efbee7" }, "amount" : 1200 },
     { "_id" : { "reportId" : "5a27cf173f978655f2efbee7", "name" : "karthik subbaraj" }, "amount" : 0 }
    ]

我想要以下结构:

{'reportid1':{'name':'sriram sathyan','amount':3000}, .....}

我尝试了以下代码:

names = defaultdict(dict)
for item in a:
   report_id = item['_id']['reportId']
   amount = item['amount']
   names[report_id] = {'name': item['_id'].get('name','')}
   if amount != 0:
      names[report_id].update({'amount': amount})
print(dict(names))

输出:

{'5a27cda63fff647c33a14b31': {'name': 'sriram sathyan'}, '5a27cf173f978655f2efbee7': {'name': 'karthik subbaraj'}}

(不是我想要的)

然后我将上面的代码更改为:

for item in a:
    report_id = item['_id']['reportId']
    amount = item['amount']
    if amount != 0:
        names[report_id] = {'amount': amount}
    names[report_id].update({'name': item['_id'].get('name','')})
print(dict(names))

输出:

{'5a27cda63fff647c33a14b31': {'amount': 3000, 'name': 'sriram sathyan'}, '5a27cf173f978655f2efbee7': {'amount': 1200, 'name': 'karthik subbaraj'}}

(我想要的!!!!!)

所以问题是=> 如何放置if语句会导致这样的改变?或者这是我在这里缺少的东西?

2 个答案:

答案 0 :(得分:2)

问题是你有相同的密钥,一个amount在数据中等于零,另一个amount 零。

在第一个循环中,首先创建一个非零amount的条目,然后被零amount的条目覆盖

因此两个代码都会覆盖某些条目,但第一个代码会覆盖您要保留的条目,而第二个代码会覆盖条目amount

答案 1 :(得分:1)

问题在于,在第一种情况下,您在此行上覆盖names[report_id]

names[report_id] = {'name': item['_id'].get('name','')}

让我们关注5a27cda63fff647c33a14b31字典中的names项:

  1. 在循环的第一次迭代中,您设置的值为names[report_id]

    {'5a27cda63fff647c33a14b31': {'name': ''}}
    
  2. 然后,由于金额为3000,字典更新为:

    {'5a27cda63fff647c33a14b31': {'name': '', 'amount': 3000}}
    
  3. 然后,在循环的第二次迭代中,字典被完全覆盖,导致完全丢失amount值:

    {'5a27cda63fff647c33a14b31': {'name': 'sriram sathyan'}}
    
  4. 使用调试器逐行执行,查看names字典的更改方式。