约束输入长度python sys.stdin.readline（）

Question

我试图从这个网站解决算法问题！ 问题是所有问题都必须通过系统输入来解决。

所以我会用

from stdin import sys
a = stdin.readline()
# do something

我的问题是有时候我需要像下面这样。 第一行是整数n （具有n位数字的1n字符串或数字的长度。

让我给你举个例子。

输入

4
1100

问题是如何限制匹配4位数字（长度）的第二行作为输入。 怎么样？ 为什么？

Answer 1

您不能以这种方式限制输入长度。

按照@Andrey的建议，您应该阅读输入，如果输入不正确，请打印错误并阅读新的输入。

顺便说一下， 在python 3中读取输入的pythonic方式是：

a = input("prompt")

因此，您可以执行以下操作：

a = ""
while len(a) != 4:
    a = input("please enter a 4 digit number: ")