파이썬과 조건문

파이썬에서는 아래 코드처럼 조건문을 작성할 수 있습니다.

if a == 1:
    print ("Hello 1")
elif a == 2:
    print ("Hello 2")
elif a == 3:
    print ("Hello 3")
elif a == 4:
    print ("Hello 4")
elif a == 5:
    print ("Hello 5")
else:
    print ("Hello Everyone")

C, C++ 또는 Java 언어와 마찬가지로 if ~ else if ~ else 문장을 지원하고 있습니다. 그렇다면 switch 문장은 어떻게 사용할까요?

자바 언어를 예로 들으면 위 문장을 아래와 같은 switch 문으로 변경할 수 있습니다.

switch (a) {
    case 1:
        System.out.println("Hello 1");
    case 2:
        System.out.println("Hello 2");
    case 3:
        System.out.println("Hello 3");
    case 4:
        System.out.println("Hello 4");
    case 5:
        System.out.println("Hello 5");
    default:
        System.out.println("Hello Everyone");
}

그런데 아쉽게도 파이썬에는 위처럼 switch 조건문이 없습니다. 꽤 아쉬운 부분이긴 합니다. 하지만 딕셔너리(Dictionary)를 사용하여 유사하게 구현할 수 있습니다.

파이썬으로 switch 흉내내기

딕셔너리(Dictionary)는 키와 데이터 값이 한 쌍을 이루는 자료형입니다. 이를 이용한다면 우리가 구현하고자 하는 switch 조건문과 유사한 기능을 사용할 수 있습니다.

# -*- coding: utf-8 -*-

def hello(a):
	return {
		1: "Hello 1",
		2: "Hello 2",
		3: "Hello 3",
		4: "Hello 4",
		5: "Hello 5",
	}.get(a, "Hello Everyone")

def main():
	print(hello(2)); # Hello 2
	print(hello(6)); # Hello Everyone

if __name__ == "__main__":
	main() 

함수(Function)와 조합해서 아래와 같이 사용도 가능합니다.

# -*- coding: utf-8 -*-

def hello1():
	print("Hello 1")

def hello2():
	print("Hello 2")	

# ... 생략

def main():
	hello_switcher = {
		1: hello1,
		2: hello2,
		3: hello3,
		4: hello4,
		5: hello5,
	}
	try:
	    # Hello 2 출력, 함수 호출
		hello_switcher[2]()
		# Hello Everyone 출력, 함수 호출
		hello_switcher[-1]()
	except Exception as e:
		print (e)
		print("Hello Everyone")

if __name__ == "__main__":
	main()

방금 살펴본 코드를 lambda를 사용하여 조금 변경할 수도 있습니다. 반환된 값이 람다 형태인지 비교하기 위해 isinstance 함수도 사용합니다.

import types

# -*- coding: utf-8 -*-

def hello1(val):
	print("Hello" + val)

def hello2(val):
	print("Hello" + val)	

# ... 생략

def hello_switcher(val):
	hello = {
		1: lambda: hello1(str(val)),
		2: lambda: hello2(str(val)),
		3: lambda: hello3(str(val)),
		4: lambda: hello4(str(val)),
		5: lambda: hello5(str(val))
	}
	helloFunc = hello.get(val, "Hello Everyone")
	
	# isinstance(a, b) : a가 b타입인가?
	if isinstance(helloFunc, types.LambdaType):
		helloFunc()
	else:
		print (helloFunc)

def main():
    # Hello 2
	hello_switcher(2)
	# Hello Everyone
	hello_switcher(6)

if __name__ == "__main__":
	main()

위에서 살펴본 것처럼 파이썬에서는 switch 문이 없어도 딕셔너리(Dictionary)를 이용해서 유사하게 구현할 수 있습니다.