import itertools

alphabet = "ЗИМА"

ar = itertools.product(alphabet, repeat=5) #Раз­ме­ще­ние с по­вто­ре­ни­ем

arl = []

for i in ar:

arl.append(list(i))

count = 0

for e in arl:

if (e.count('И') == 1 and e.count('А') == 0) or (e.count('И') == 0 and e.count('А') == 1):

count += 1

print(count)

from itertools import product

words = [''.join(i) for i in product(sorted('ЗИМА'), repeat=5)]

count = 0

for i in range(len(words)):

if words[i].count('И') + words[i].count('А') == 1:

count += 1

print(count)