输入股票代码判断深市还是沪市 python

gp='002340'# 沪市股票包含上证主板和科创板和B股：沪市主板股票代码是60开头、科创板股票代码是688开头、B股代码900开头。# 深市股票包含主板、中小板、创业板和B股：深市主板股票代码是000开头、中小板股票代码002开头、创业板300开头、B股代码200开头def gp_type_szsh(gp):if gp.find('60',0,3)==0:gp_type='sh'elif gp.find('688',0,4)==0:gp_type='sh'elif gp.find('900',0,4)==0:gp_type='sh'elif gp.find('00',0,3)==0:gp_type='sz'elif gp.find('300',0,4)==0:gp_type='sz'elif gp.find('200',0,4)==0:gp_type='sz'return gp_typeprint(gp_type_szsh(gp))

最近在搞股票写爬虫，感觉这里可以优化一下，上网找了一下没有资源就自己写一个分享出来。