编写函数，以字典形式返回所需的股票数量

编写函数，接受任意数量的股票代码，以字典形式返回代码中沪市主板和深市主板股票数量。沪市主板的股票代码以60开头，深市主板的股票代码以000开头

def stocknum():
    adi={};result={}
    a=input('请输入股票代码(逗号隔开):')
    if len(a)==6:
        if a[0:2]=='000':
            adi['沪市股票数']=1
        else:
            adi['深市股票数']=1    
    else:
        atu=tuple(a.split(','))
        for i in atu:
            if i[0:3]!='000' and i[0:3]!='600':
                print('请输入正确股票代码！')
            else:
                result[i[0:3]]=result.get(i[0:3],0)+1
                adi['沪市股票数']=result.get('000');adi['深市股票数']=result.get('600')

    return adi
stocknum()

数据源呢？要自己爬吗？