算法题字符串相关

 磁盘的容量单位常用的有M， G， T这三个等级，它们之间的换算关系为1T = 1024G， 1G = 1024M，现在给定n块磁盘的容量，请对它们按从小到大的顺序进行稳定排序，例如给定5 块盘的容量， 1T， 20M， 3G， 10G6T， 3M12G9M排序后的结果为20M， 3G， 3M12G9M， 1T， 10G6T。注意单位可以重复出现，上述3M12G9M表示的容量即为3M+12G+9M，和 12M12G相等。 输入描述： 输入第一行包含一个整数n(2 <= n <= 100)，表示磁盘的个数，接下的n行，每行一个字符串 (长度大于2，小于30)，表示磁盘的容量，由一个或多个格式为mv的子串组成，其中m表示 容量大小， v表示容量单位，例如20M， 1T， 30G， 10G6T， 3M12G9M

def str_change(disk_str):
    str_size = {"M": 1, "G": 1024, "T": 1024*1024, }
    in_size = 0
    fin_size = 0
    for i in disk_str:
        if i.isdigit():
            in_size = 10*int(in_size) + int(i)
        else:
            if i in str_size:
                fin_size = in_size*str_size[i] + fin_size
                in_size = 0
            else:
                return EOFError
    return fin_size

def sort_disk(disk_list):
  return sorted(disk_list ,key=lambda x: str_change(x))


if __name__ == "__main__":
    n = int(input())
    disk_list = []
    for i in range(n):
        disk_str = input()
        disk_list.append(disk_str)
    print(sort_disk(disk_list))