有无python巨佬帮看看 ，功能已经全部实现，卡在输入格式了

问题是： 要按2次回车，一次作为结束标志，一次作为执行计算, 所以测试环境过不了，哎

下面是代码：

def count_zeros_and_ones(result_str):
    count = 1         # 当前数字，记录它本身的个数
    out_str = ""  # 用来存放每一部分连续的数量
    for i in range(1, len(result_str)):
        if result_str[i] == result_str[i - 1]:
            count += 1
        else:          # 变为另一个数字
            out_str += str(count)  # ！存完再变count
            count = 1  # 记录它本身的个数
    # 最后一组不会变0或1了，记得补充上
    out_str += str(count)
    return out_str
matrix = []

while True:
    row = input()
    if row.strip() == "":  # 判断是否输入了空行(换行)
        break
    matrix.append(row)

n = len(matrix)
result_str = ''.join(matrix)
# print(' '.join(f"{str(n)}" + count_zeros_and_ones(result_str)))
print(' '.join(f"{str(n)}{count_zeros_and_ones(result_str)}"))