如何修改Python代码以获取符合条件的输出结果数量？

你可以通过引入计数器变量来统计符合条件的数字数量，修改后的代码如下：

原代码 #

for i in range(100, 1000):
    s = str(i)
    k1 = int(s[0]) + int(s[1])
    k2 = int(s[1]) + int(s[2])
    sd = str(max(k1, k2))
    ft = str(min(k1, k2))
    s1 = ft + sd
    if s1 == '1216':
        print(i)

修改后的代码（统计数量） #

count = 0
for i in range(100, 1000):
    s = str(i)
    k1 = int(s[0]) + int(s[1])
    k2 = int(s[1]) + int(s[2])
    sd = str(max(k1, k2))
    ft = str(min(k1, k2))
    s1 = ft + sd
    if s1 == '1216':
        count += 1
print(count)

说明 #

• 初始化count为0，用于记录符合条件的数字个数
• 每找到一个满足s1 == '1216'的数字，就将count加1
• 循环结束后打印count，即可直接得到结果数量6

如果你追求更简洁的写法，也可以用生成器表达式实现：

count = sum(
    1 for i in range(100, 1000)
    if str(min(int(str(i)[0]) + int(str(i)[1]), int(str(i)[1]) + int(str(i)[2]))) +
       str(max(int(str(i)[0]) + int(str(i)[1]), int(str(i)[1]) + int(str(i)[2]))) == '1216'
)
print(count)

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