Python——sorted函数练习

(2018-携程-春招题)题目需求:
给定一个整形数组， 将数组中所有的0移动到末尾， 非0项保持不变；
在原始数组上进行移动操作， 勿创建新的数组;
输入：
第一行是数组长度， 后续每一行是数组的一条记录;
4
0
7
0
2
输出：
调整后数组的内容;
4
7
2
0
0

代码如下：

a = ''.join(input().split())    #将输入的字符串分隔，连接转换整形列表
li = [int(i) for i in a]

def f(x):
    if x == 0:
        return 1
    else:
        return 0
print(sorted(li,key=f))    #默认按照值由小到大排序，非0的数默认大小为0

如果可以新建列表，实现代码如下：

str = input()
alist = []
blist = []
for i in str:
    if int(i) != 0:
        alist.append(int(i))
str_len = len(str)
alist_len = len(alist)
for i in range(str_len-alist_len):
    blist.append(0)
print(alist+blist)