以下代码不模拟输入,是否可以使用生成器来模拟输入?
inputlist = [5,1,2,3,4,5,6]
def mock_input(dummy_prompt=">>> "):
global inputlist
for u in inputlist:
yield str(u)
__builtins__.input = mock_input
l=[]
x= input()
for u in range(int(x)):
l.append(int(input()))
if l==inputlist[1:]:
print("bravo")
错误是x是生成器而不是String。
我必须使用全局变量创建自己的生成器,还是错过语法中的某些内容。
您要分配一个生成器,__builtins__.input
而不是使用生成器进行迭代并返回一个值,这就是为什么它不起作用的原因。
而是使用创建一个generateiter
并使用它进行迭代next(..)
inputlist = [6,1,2,3,4,5,6]
gen = iter(inputlist)
def mock_input(dummy_prompt=">>> "):
return str(next(gen))
__builtins__.input = mock_input
l=[]
x= input()
for u in range(int(x)):
l.append(int(input()))
if l==inputlist[1:]:
print("bravo")
PS:我将列表的第一个数字修改为6,否则您丢失了一个项目,并且bravo没有打印
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句