In [ ]:

    

in_file = open('in.txt')
out_file = open("out.txt", 'w')
while True:
    lines = in_file.readline() # 整行读取数据
    if not lines:
        break
    #print (lines)
    if lines[0] == '0':
        #print(lines[:6]+".XSHE")
        out_file.write(lines[:6]+".XSHE"+'\n')
    if lines[0] == '3':
        #print(lines[:6]+".XSHE")
        out_file.write(lines[:6]+".XSHE"+'\n')
    if lines[0] == '6':
        #print(lines[:6]+".XSHG")
        out_file.write(lines[:6]+".XSHG"+'\n')
in_file.close()
out_file.close()