Вместо написания длинных циклов
for с созданием пустого списка и условиями if, используйте List Comprehensions. Они создают новые списки в одну строку кода. К тому же, такие конструкции работают быстрее обычных циклов на уровне самого интерпретатора.Разъяснивший Python