Подсчёт комбинаций: Определите количество девятеричных пятизначных чисел; 0 · ЕГЭ №8

from itertools import product
digits = '012345678'
res = []
for i in product(digits, repeat=5):
    if i[0] == '0': continue
    if i.count('0') != 1: continue
    zero_pos = i.index('0')
    flag = True
    if zero_pos > 0 and int(i[zero_pos-1]) % 2 == 1: flag = False
    if zero_pos < 4 and int(i[zero_pos+1]) % 2 == 1: flag = False
    if flag: res.append(i)
print(len(res))