基于条件的循环范围

Question

我想说的是：

if switch == True:
    for i in range(5):
        # stuff()
else:
    for i in range(10):
        # stuff()

有更多的Pythonic方式吗？

Answer 1

将其与conditional expression：

组合在一起

for i in range(5 if switch else 10):
    # do something with i

您无需在此处测试== True; if 已经为您完成此操作。

更具可读性的是将结束值分离为变量：

end = 5 if switch else 10
for i in range(end):
    # do something with i

Answer 2

使用if/else的内联简短形式：

for i in range(5 if switch else 10):
    # stuff()