2017-09-21 1 views
0
dict = {a:[1,2,3,4], b:[4,5,6]} 
내가 인수

Python의 요소 사전에 Or 조건을 작성하려면 어떻게합니까?

def func1(dict, elem1, elem2): 

로 사전에 걸리는 함수에서 func1를 작성하고 간단한 방법을 사용하고 싶습니다

은 기본적으로 다음과 같은 작성하는 조건

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

사전에 할 수있는 경우 복수의 키 - 값 엔트리가 있음

답변

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

?

은 yeild 것 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 

다음 ...이 구현은 거짓

을 반환
관련 문제