Python循环未按预期在用户输入后终止问题求助

我想写一个for循环给列表添加元素，当列表总元素数达到5时终止。但实际运行时，列表元素数到6才停止，本该只请求2次用户输入，结果请求了3次。以下是我的代码：

# step 1
beatles = []
print("Step 1:", beatles)

# step 2
beatles.append("John Lennon")
beatles.append("Paul McCartney")
beatles.append("George Harrison")
print("Step 2:", beatles)

# step 3
for i in range(len(beatles)):
    if i == 5:
        break
    beatles.append(input("Enter the name of a beatles: "))
print("Step 3:", beatles)

问题原因 #

• 用range(len(beatles))创建循环时，len(beatles)在循环启动瞬间就被计算为3（此时列表已有3个成员），所以循环会固定执行3次（i依次取0、1、2）。
• 终止条件if i ==5: break完全无效，因为i的最大值只有2，根本达不到5，因此循环会完整执行3次append操作，最终列表元素数变为3+3=6。

解决方案 #

方案1：用while循环（更直观） #

直接判断列表长度是否小于5，满足条件就继续添加元素：

# step 3
while len(beatles) < 5:
    beatles.append(input("Enter the name of a beatles: "))
print("Step 3:", beatles)

方案2：用for循环执行固定次数 #

先计算需要添加的元素数量（目标总数5减去当前列表长度3，得到2），直接循环对应次数：

# step 3
need_add = 5 - len(beatles)
for _ in range(need_add):
    beatles.append(input("Enter the name of a beatles: "))
print("Step 3:", beatles)

内容的提问来源于stack exchange，提问作者Sabino Alcala