Объединение для и если петли

Question

У меня есть код:

num_for_loops = 0 
for line in lines: 
    if line.strip().startswith('for '): 
     num_for_loops += 1 
return num_for_loops 

Мне нужно уплотнить его вниз как черточки, как это возможно. Есть ли способ сделать это, объединив утверждения for и if? Контекст не имеет значения.

Answer 1

num_for_loops = sum(line.strip().startswith("for ") for line in lines) 

, так как вы сразу же возвращаетесь num_for_loops Я полагаю, вы на самом деле просто хотите

return sum(line.strip().startswith("for ") for line in lines) 

Answer 2

Генератора выражения всегда выигрывают краткий код конкурсы:

return sum(1 for line in lines if line.strip().startswith('for ')) 

Это порождает ряд 1s равного в длина к числу строк, начинающихся с 'for', суммируя список по мере его создания, чтобы вы могли анализировать терабайты данных без нехватки памяти.