我有一本字典,它有一个键(一个两位数的前缀)和一个值(一个字符串列表)。看起来像这样:
prefixes = {
"AC" : ["england-wales"],
"OC" : ["england-wales", "wales"],
"SC" : ["scotland"],
"NC" : ["northern-ireland"]
}
我正在尝试创建一个前缀列表,其中值包含列表中提供的区域。
通过的区域可以是一个或多个区域,例如:
regions = ['england-wales', 'wales']
因此,利用以上数据,我想创建一个称为前缀的新列表,它将包含所有前缀,其中值中包含 'wales'或'england-wales':
prefix_list = ['AC', 'OC']
如果区域等于“威尔士”,则:
prefix_list = ['OC']
我已经尝试了一些事情,并得到了这个凌乱的解决方案:
combo_list = []
prefix_list = []
for r in regions:
combo_list.append([k for k, v in prefixes.items() if r in v])
for sublist in combo_list:
for i in sublist:
prefix_list.append(i)
prefix_list = sorted(set(prefix_list)) #This will remove duplicates for me
我认为必须有一种更好,更“干净”的方式来做到这一点,但还不能完全弄清楚。任何帮助表示赞赏。
答案 0 :(得分:6)
为此,您可以使用any()
-作为列表理解:
prefix_list = [k for k, v in prefixes.items() if any(x in v for x in regions)]
给我们
>>> prefix_list
['AC', 'OC']