我有一套Python
A = {34, 4, 7, 39, 40, 13, 18, 20}
和一本字典
B = {4: [6, 34, 39], 7: [18, 34], 13: [6, 18, 34], 18: [7, 13],
20: [31, 47], 34: [4, 7, 13], 39: [4, 8], 40: [15, 43]}
我想创建一个新的字典C,以便如果B中的任何值在A中包含一个数字,则必须将它们从相应的值列表中删除。按键应不受影响。
例如,我正在寻找的理想输出是:
C = {4: [6], 7: [], 13: [6], 18: [], 20: [31, 47], 34: [], 39: [8]}
我尝试使用以下代码解决此问题:
C = {}
for key, value in B_mod.items():
if value in A:
C[key] = value
print("C is:", C)
我得到的输出如下:
C is: {7: 18, 18: 7, 34: 4, 39: 4}
显然,它不起作用。
*请注意,这B_mod
意味着我正在使用以下代码在字典B的值中取消列出该列表:
B_mod = {k:v[0] for k,v in B.items()} # Unlisitng the values of dictionary B
我这样做是因为我无法遍历列表作为B中的值。*
有帮助获得C吗?
您绝对可以遍历列表中的值B
,的确确实需要,所以像这样:
>>> A = {34, 4, 7, 39, 40, 13, 18, 20}
>>> B = {4: [6, 34, 39], 7: [18, 34], 13: [6, 18, 34], 18: [7, 13], 20: [31, 47], 34: [4, 7, 13], 39: [4, 8], 40: [15, 43]}
>>> C = {}
>>> for key, value in B.items():
... new_value = []
... for val in value:
... if val not in A:
... new_value.append(val)
... C[key] = new_value
...
>>> C
{4: [6], 7: [], 13: [6], 18: [], 20: [31, 47], 34: [], 39: [8], 40: [15, 43]}
>>>
您还可以将字典理解与列表理解一起使用:
>>> {k:[v for v in vs if v not in A] for k, vs in B.items()}
{4: [6], 7: [], 13: [6], 18: [], 20: [31, 47], 34: [], 39: [8], 40: [15, 43]}
这等效于for循环方法。对您更有意义的那个都可以。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句