输入股票代码判断深市还是沪市，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_type
print(gp_type_szsh(gp))

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