python 中無switch（寫了幾個函數代替switch）

字典的常用用途之一代替switch

在C/C++/Java語言中，有個很方便的函數switch，比如:

復制代碼代碼如下:

public class test {  
      
    public static void main(String[] args) {  
        String s = "C";  
        switch (s){  
        case "A":   
            System.out.println("A");  
            break;  
        case "B":  
            System.out.println("B");  
            break;  
        case "C":  
            System.out.println("C");  
            break;  
        default:  
            System.out.println("D");  
        }  
    }  
}  

 

在Python中要實現同樣的功能，
方法一，就是用if, else語句來實現，比如:

from __future__ import division  
  
def add(x, y):  
    return x + y  
  
def sub(x, y):  
    return x - y  
  
def mul(x, y):  
    return x * y  
  
def div(x, y):  
    return x / y  
  
def operator(x, y, sep='+'):  
    if   sep == '+': print add(x, y)  
    elif sep == '-': print sub(x, y)  
    elif sep == '*': print mul(x, y)  
    elif sep == '/': print div(x, y)  
    else: print 'Something Wrong'  
  
print __name__  
   
if __name__ == '__main__':  
    x = int(raw_input("Enter the 1st number: "))  
    y = int(raw_input("Enter the 2nd number: "))  
    s = raw_input("Enter operation here(+ - * /): ")  
    operator(x, y, s)  

 

方法二，用字典來巧妙實現同樣的switch的功能，比如:

復制代碼代碼如下:

#coding=gbk  
  
from __future__ import division  
  
x = int(raw_input("Enter the 1st number: "))  
y = int(raw_input("Enter the 2nd number: "))  
  
def operator(o):  
    dict_oper = {  
        '+': lambda x, y: x + y,  
        '-': lambda x, y: x - y,  
        '*': lambda x, y: x * y,  
        '/': lambda x, y: x / y}  
    return dict_oper.get(o)(x, y)  
   
if __name__ == '__main__':    
    o = raw_input("Enter operation here(+ - * /): ")  
    print operator(o)