1的个数排序,从小到大输出

对于给定若干由0、1构成的字符串长度不超过80,要求将它们按1的个数从小到大排序。若1的个数相同,则按字符串本身从小到大排序。 要求至少使用一个自定义函数。
输出
对于每组测试,输出排序后的结果,每个字符串占一行。
输入
3
10011111
00001101
1010101
输出
00001101
1010101
10011111

def my_sort(l):
    return sorted(l, key = lambda x: (x.count('1'), x))

n = int(input(">>>"))
l = []
for i in range(n):
    l.append(input(">>>"))

for i in my_sort(l):
    print(i)

count = int(input())
lst = []
count_1 = []
for i in range(count):
    lst.append(input())

sorted(lst , key = lambda x: (x.count('1'), x))

for i in lst:
    print(i)