api接口做股票量化分析时,股票代码要加点sh或者点sz,所以这里就需要做一个判断后,然后加上后缀,判断的方法有很多,这里我用的是正则表达式的match()函数
代码如下:
list1=["000326","300456","688456"]
for daima in list1:
result=re.match(r"\d",daima) #用正则的match匹配第一个数字,然后做判断
# print(result.group())
if int(result.group())==0 or int(result.group())==3:
print(daima+".SZ")
else:
print(daima+".SH")