Python代码最后一个for循环仅处理首个元素问题求助

问题描述 #

编写的Python代码中，最后一个for循环仅对列表的第一个元素执行操作，其余元素未被处理。已尝试Python 3和PyPy 3环境，替换eval()为exec()后问题仍存在。

原代码 #

a = int(input())
setA = set(map(int, input().split()))
n = int(input())
com = [input() for alnum in range(n*2)]
nestcom = []
cmd = []
for i in com:
    if com.index(i)%2 == 0:
        nestcom.append([i, com[com.index(i)+1]])
    else:
        continue
for x in nestcom:
    y, z = x[0].split(), set(map(int, x[1].split()))
    cmd.append([y, z])
for u in cmd:
    eval('setA.'+cmd[0][0][0]+'('+str(cmd[0][1])+')')
print(sum(setA))

测试输入 #

16
1 2 3 4 5 6 7 8 9 10 11 12 13 14 24 52
4
intersection_update 10
2 3 5 6 8 9 1 4 7 11
update 2
55 66
symmetric_difference_update 5
22 7 35 62 58
difference_update 7
11 22 35 55 58 62 66

问题根源 #

1. 硬编码索引导致循环失效：最后一个循环中，虽然用u遍历cmd列表，但执行语句始终调用cmd[0]（即第一个命令），完全没有使用当前循环的u变量，导致所有迭代都重复执行第一个命令，后续命令被直接忽略。
2. com.index(i)的潜在错误：如果com列表存在重复元素，index()会返回第一个匹配项的索引，导致命令与参数错误关联，生成的nestcom列表逻辑出错。

修正后的代码 #

a = int(input())
setA = set(map(int, input().split()))
n = int(input())
com = [input().strip() for _ in range(n*2)]
nestcom = []
cmd = []

# 用enumerate遍历索引，避免index()的重复元素问题
for idx, item in enumerate(com):
    if idx % 2 == 0:
        nestcom.append([item, com[idx+1]])

for x in nestcom:
    cmd_parts, num_set = x[0].split(), set(map(int, x[1].split()))
    cmd.append([cmd_parts, num_set])

# 使用当前循环变量u，而非硬编码cmd[0]，用getattr替代eval更安全
for u in cmd:
    method_name = u[0][0]
    method_args = u[1]
    getattr(setA, method_name)(method_args)

print(sum(setA))

关键修正点 #

• 替换com.index(i)为enumerate遍历索引，确保命令与参数正确配对。
• 最后循环中使用当前迭代的u变量获取命令和参数，取消硬编码的cmd[0]调用。
• 用getattr()替代eval()，既规避字符串拼接执行代码的安全风险，也让代码逻辑更清晰。

内容的提问来源于stack exchange，提问作者Ackumey Joseph