如何在Python中的元素字典上创建Or条件？

Question

dict = {a:[1,2,3,4], b:[4,5,6]}

我想创建一个函数func1，它将字典作为参数

def func1(dict, elem1, elem2):

并使用一种简单的方法基本上创建以下条件

if ((elem1 ==a and elem2 in dict[a]) or (elem1 == b and elem2 in dict[b]))

字典可以有多个键值条目

Answer 1

def func1(a_dict, elem1, elem2):
    return elem2 in a_dict.get(elem1,[])

我想？

会提供以下内容

data = {'a':[1,2,3,4], 'b':[4,5,6]}

print(func1(data,'a',6)) #False
print(func1(data,'a',2)) #True
print(func1(data,'b',2)) #False
print(func1(data,'b',6)) #True

虽然如果elem1不在字典中，我不确定你的期望...这个实现只返回False