平方数字函数挑战不起作用

Question

难点在于输入数字的平方。

import sys
# import numpy as np
# import pandas as pd
# from sklearn import ...
def square(x):
return print(x ** 2)
for line in sys.stdin:
print(line, end="")

我不知道如何使用sys。stdin我猜是因为我不知道如何通过x的挑战测试输入。

对不起，我找不到任何答案。

Answer 1

如果要从标准输入中读取行，可以使用input。它将在文件末尾引发EOFError。要连续读取行并处理它们，您可以使用EOFError的异常处理程序在循环中读取。

try:
while True:
line = input()
x = int(line)
print(square(x))
except EOFError:
pass

或者为了好玩，可以将其封装在一个函数中，生成以下行。

def read_stdin():
try:
while True:
yield input()
except EOFError:
pass
for line in read_stdin():
print(square(int(line)))