python - 如何使用 if-condition 以不同的迭代样式执行相同的命令？（Python，代码优化）

Question

所以，我有这个代码：

    t = PrettyTable(['first', 'second'])
    if condition:
        for i in data_set[::-1]:
            t.add_row([i['first'], i['second'])
            # ... (multiple other commands)
    else:
        for i in data_set:
            t.add_row([i['first'], i['second'])
            # ... (same multiple commands as above)

我想缩短代码，这样我只需要编写一次命令。那可能吗？

标签： pythonpython-3.xif-statementoptimizationprettytable

Answer 1

只需使用条件来获得正确的迭代然后做你的事情

items = data_set if condition else data_set[::-1]

for i in items:
    t.add_row([i['first'], i['second'])