检查字典值是否包含列表项

时间:2020-01-23 15:49:45

标签: python list dictionary for-loop list-comprehension

我有一本字典,它有一个键(一个两位数的前缀)和一个值(一个字符串列表)。看起来像这样:

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

我认为必须有一种更好,更“干净”的方式来做到这一点,但还不能完全弄清楚。任何帮助表示赞赏。

1 个答案:

答案 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']