三角系列数字的用户输入

Question

   def my_func(c=1, n=None):
    while True:
        if c > n:
            # if counter > number of iterations required
            break
        a = int(c * (c + 1)/2)
        yield a
        c += 1

my_object = my_func(n=5)

while True:
    try:
        print(next(my_object))
    except StopIteration:
        print("End of Iteration")
        break
   

您好，我有以下内容，基本上它打印三角形数字，我想做的是设置n=用户键盘输入，基本上用户输入一个迭代，它打印出输入的迭代次数的三角形序列。

如果我想在n=无时无限循环，那么这大概就是n=“”。。。但我不知道怎么做

任何帮助都将不胜感激。

这是Python编码

使用Seons解决方案时出错

错误！请输入数字或留空。回溯（最近一次调用）：文件""，第15行，get_valid_user_inputValueError：以10为底的int（）的无效文字：'a'

在处理上述异常时，发生了另一个异常：

Traceback（最近一次调用）： File""， line 20， in File""， line 18， inget_valid_user_inputNameError： name'get_user_input'不是

Answer 1

我们从稍微修改my\u func开始，如果n

• 要么他什么都不输入（空字符串），在这种情况下，我们假设他想无限期循环并将n设置为-1
• 或者他输入了其他内容，在这种情况下，我们假设他输入了一个整数。然后我们使用int（user_input）将此值从str转换为andint，并将其用作my_func
的n参数

def my_func(c=1, n=-1):
    while True:
        if n>-1 and c > n:
            # if counter > number of iterations required
            break
        a = int(c * (c + 1)/2)
        yield a
        c += 1

user_input = input("Enter a number, or leave blank for infinite loop: ")
if user_input == "":
    n = -1
else:
    n = int(user_input)
    
my_object = my_func(n=n)

while True:
    try:
        print(next(my_object))
    except StopIteration:
        print("End of Iteration")
        break

下面给出了一个更复杂的版本，它确保用户输入有效。我们递归调用get\u valid\u user\u input函数，直到用户输入为空字符串或整数。

def my_func(c=1, n=-1):
    while True:
        if n>-1 and c > n:
            # if counter > number of iterations required
            break
        a = int(c * (c + 1)/2)
        yield a
        c += 1

def get_valid_user_input():
    user_input = input("Enter a number, or leave blank for infinite loop: ")
    if user_input == "":
        return -1
    try:
        return int(user_input)
    except ValueError:
        print("Error! Please enter a number or leave blank.")
        return get_valid_user_input()

my_object = my_func(n=get_valid_user_input())
while True:
    try:
        print(next(my_object))
    except StopIteration:
        print("End of Iteration")
        break