Как выполнять те же команды, но в разных стилях итерации с условием if? (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)

, и я хочу сократить код, чтобы мне приходилось писать команды только один раз. Это возможно?

Answer 1

Просто используйте условие, чтобы получить правильную итерацию, затем делайте ваши вещи

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

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