从具有二进制值的字典生成Pair



我有一个这样的输入字典:

dict1= {'AM': ['tv', 'rs', 'pq', 'MN', 'tN', 'tq', 'OP', 'tP', 'QR', 'tr'],
'BR': ['tv', 'rs', 'pq', 'MN', 'tN', 'tq', 'OP', 'tP', 'QR', 'tr']}

我要查找的另外两个字典

dict2={'AM':{'pq','rs','tv'},'BR':{'MN','OP','QR'}}
dict3={'AM':{'tq','rs','tv'},'BR':{'tN','tP','tr'}}

预期输出

{'AM-tv': 1,
'AM-rs': 1,
'AM-pq': 1,
'AM-MN': 0,
'AM-tN': 0,
'AM-tq': 1,
'AM-OP': 0,
'AM-tP': 0,
'AM-QR': 0,
'AM-tr': 0,
'BR-tv': 0,
'BR-rs': 0,
'BR-pq': 0,
'BR-MN': 1,
'BR-tN': 1,
'BR-tq': 0,
'BR-OP': 1,
'BR-tP': 1,
'BR-QR': 1,
'BR-tr': 1}

在输出中,需要从dict1生成对。如果dic1键的值出现在dic2或dic3键的值中,那么它在输出中的值将为1。如果值不存在,那么它在新字典中的值将为0。有办法吗?

可以使用两个简单的嵌套循环。

这是一个应该在大多数python版本中工作的变体:

out = {}
dicts = [dict2, dict3]
for k,l in dict1.items():
for v in l:
out[f'{k}-{v}'] = 1 if any(v in d[k] for d in dicts) else 0

或者,作为字典理解式:

out = {f'{k}-{v}': 1 if any(v in d[k] for d in dicts) else 0
for k,l in dict1.items() for v in l}

输出:

{'AM-tv': 1,
'AM-rs': 1,
'AM-pq': 1,
'AM-MN': 0,
'AM-tN': 0,
'AM-tq': 1,
'AM-OP': 0,
'AM-tP': 0,
'AM-QR': 0,
'AM-tr': 0,
'BR-tv': 0,
'BR-rs': 0,
'BR-pq': 0,
'BR-MN': 1,
'BR-tN': 1,
'BR-tq': 0,
'BR-OP': 1,
'BR-tP': 1,
'BR-QR': 1,
'BR-tr': 1}

一种选择是在遍历dict1时对dict1列表中的每个元素进行集合成员检查:

out = {}
for k, lst in dict1.items():
s =  dict2[k] | dict3[k]
for v in lst:
out[f"{k}-{v}"] = (v in s) * 1

我们也可以编写与字典推导式相同的代码(但它比另一种选择慢,因为我们设置联合len(lst)次,而不是在循环中设置一次):

out = {f"{k}-{v}": (v in dict2[k] | dict3[k]) * 1 for k, lst in dict1.items() for v in lst}

输出:

{'AM-tv': 1,
'AM-rs': 1,
'AM-pq': 1,
'AM-MN': 0,
'AM-tN': 0,
'AM-tq': 1,
'AM-OP': 0,
'AM-tP': 0,
'AM-QR': 0,
'AM-tr': 0,
'BR-tv': 0,
'BR-rs': 0,
'BR-pq': 0,
'BR-MN': 1,
'BR-tN': 1,
'BR-tq': 0,
'BR-OP': 1,
'BR-tP': 1,
'BR-QR': 1,
'BR-tr': 1}

您可以通过首先为每个键创建dict2dict3中存在的所有值的集合,然后检查dict1中存在的所有键和值的组合来实现您想要的结果:

present = { k : dict2[k] | dict3[k] for k in dict2 }
result = { f'{k}-{v}' : (v in present[k]) * 1 for k in dict1.keys() for v in dict1[k] }

输出:

{
'AM-tv': 1, 'AM-rs': 1, 'AM-pq': 1, 'AM-MN': 0, 'AM-tN': 0,
'AM-tq': 1, 'AM-OP': 0, 'AM-tP': 0, 'AM-QR': 0, 'AM-tr': 0,
'BR-tv': 0, 'BR-rs': 0, 'BR-pq': 0, 'BR-MN': 1, 'BR-tN': 1,
'BR-tq': 0, 'BR-OP': 1, 'BR-tP': 1, 'BR-QR': 1, 'BR-tr': 1
}

注意-将布尔值乘以1与使用int(...)将其转换为int相同,但速度更快。

相关内容

  • 没有找到相关文章

最新更新