如何使用不同于按功能输入和排序列表的哨兵值?

时间:2019-10-04 23:39:33

标签: python list while-loop

我正在学习初学者python,我遇到了一个问题。

问题涉及询问用户输入的蘑菇量,输入重量,然后根据用户输入对蘑菇进行分类。为此,需要一个列表和while循环将输入追加到列表中。

当前,我正在尝试实现一个哨兵值,该值将在输入所有用户输入后停止while循环,但是将哨兵设置为“ STOP”会与int()通知冲突。

if __name__ == "__main__":
    STOP = "stop"
    mushroom = int(input("Enter a mushroom weight in grams, or STOP to end. "))
    total_list = []
    while total_list != STOP:
        total_list.append(mushroom)
        mushroom = int(input("Enter a mushroom weight in grams, or STOP to end. "))
    print(total_list)

程序运行良好,直到输入“ STOP”,出现语法错误。

mushroom = int(input("Enter a mushroom weight in grams, or STOP to end. "))
ValueError: invalid literal for int() with base 10: 'STOP'

如您所见,前哨值STOP与我的输入建议冲突,从而导致错误。

对于问题的第二部分,我需要按权重对输入的值进行排序。如果一切都正确完成,我应该列出所有值。 我可以使用哪种代码对值进行排序? 我需要基于小(<100),中(100-1000)和大(> 1000)对每个整数值进行排序,然后在语句中打印结果。 我在这里需要做什么有点不知所措。

谢谢,只是停留在一个地方。

2 个答案:

答案 0 :(得分:0)

第一次崩溃是因为您试图将“ STOP”转换为整数。例如,如果您打开python解释器并键入int(“ STOP”),则将发生相同的崩溃。解释器不知道如何将字符串“ STOP”转换为整数。更好的方法可能是检查是否可以使用isdigit()或try / except将输入字符串转换为整数。

对于过滤,最简单的方法是在列表上使用列表理解。这样的事情应该起作用:

if __name__ == "__main__":
    mushroom = None
    total_list = []
    while mushroom != "STOP":
        mushroom = input("Enter a mushroom weight in grams, or STOP to end. ")
        if mushroom.isdigit():
            total_list.append(int(mushroom))
    sorted_total_list = sorted(total_list)
    small_values = [mushroom for mushroom in total_list if mushroom < 100]
    medium_values = [mushroom for mushroom in total_list if 100 <= mushroom <= 1000]
    large_values = [mushroom for mushroom in total_list if mushroom > 1000]
    print(sorted_total_list)
    print(small_values)
    print(medium_values)
    print(large_values)

答案 1 :(得分:0)

您可以使用在Python中使用tryexcept的优势,尝试将非整数转换为整数而产生的错误。这是documentation

重写代码,您可以尝试以下操作:

if __name__ == "__main__":
    STOP = "STOP"
    total_list = []
    while True:
        try:
            user_input = input("Enter a mushroom weight in grams, or STOP to end. ")
            total_list.append(int(user_input))  
        except ValueError:
            if user_input.strip().upper()==STOP:
                break
            else:
                print("Incorrect entry! '{}' is not an integer".format(user_input))     

然后,如果您想对该列表进行排序并分类,则可以考虑使用字典:

total_list.sort()       
di={'Small':[],'Medium':[],'Large':[]}
for e in total_list:
    key='Small' if e<100 else 'Medium' if 100<=e<=1000 else 'Large'
    di[key].append(e)

print(total_list, sum(total_list),di)