如何补足股票代码0开头的，以及给代码添加后缀?

Question

如何补足股票代码0开头的，以及给代码添加后缀：0、3开头的股票代码添加 .SZ，6开头添加 .SH?
原代码：已经把50个CSV里的收盘价汇总到了一个总表，但第一行显示的股票代码缺零，缺后缀。

这是汇总了的CSV内容：第一行缺零，缺后缀

这是单个CSV的内容：

Answer 1

使用遍历代码列，字符串拼接及zfill函数操作即可。参考代码如下：

import pandas as pd
df=pd.read_excel('stock_002254.xlsx')
new=[]
for d in df['code']:
    if len(str(d))<6:
        new.append(str(d).zfill(6)+'.SZ')
    elif str(d).startswith('3')and len(str(d))==6:
        new.append(str(d)+'.SZ')
    elif str(d).startswith('6')and len(str(d))==6:
        new.append(str(d)+'.SH')
    else:
        new.append(str(d))
df['code']=new
print(df)
df.to_csv('stock_002254_1.csv',index=False)