我有以下内容:
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语句会导致这样的改变?或者这是我在这里缺少的东西?
答案 0 :(得分:2)
问题是你有相同的密钥,一个amount
在数据中等于零,另一个amount
不零。
在第一个循环中,首先创建一个非零amount
的条目,然后被零amount
的条目覆盖
因此两个代码都会覆盖某些条目,但第一个代码会覆盖您要保留的条目,而第二个代码会覆盖条目amount
。
答案 1 :(得分:1)
问题在于,在第一种情况下,您在此行上覆盖names[report_id]
:
names[report_id] = {'name': item['_id'].get('name','')}
让我们关注5a27cda63fff647c33a14b31
字典中的names
项:
在循环的第一次迭代中,您设置的值为names[report_id]
:
{'5a27cda63fff647c33a14b31': {'name': ''}}
然后,由于金额为3000,字典更新为:
{'5a27cda63fff647c33a14b31': {'name': '', 'amount': 3000}}
然后,在循环的第二次迭代中,字典被完全覆盖,导致完全丢失amount
值:
{'5a27cda63fff647c33a14b31': {'name': 'sriram sathyan'}}
使用调试器逐行执行,查看names
字典的更改方式。