关于股票代码转换问题

请问怎样将每个股票代码转换，例如sh600001，转为600001.sh这样？

如果格式都是前两个是字母后面是数字的话

new_code = code[2:] + '.' + code[:2]