ArcPy-if语句

def Cal(value):
  if(value=='YSP'):
    return 1;
  elif(value=='WSP'):
    return 2;
  elif(value=='JSP'):
    return 3;

 如果是两个参数：

def Cal(value1,value2):
  if(value2==1):
    return 'YS'+value1;
  elif(value2==2):
    return 'WS'+value1;
  elif(value2==3):
    return 'JS'+value1;