import itertools N = 12 # počet baterií V = 4 # počet vybitých baterky = [0]*V + [1]*(N-V) celkem = 0 # celkový počet možností a = 0 # kolik z nich splňuje (a) b = 0 # kolik z nich splňuje (b) for vyber in itertools.combinations(baterky, 3): celkem += 1 pocet_ok = sum(vyber) if pocet_ok == 3: a += 1 if pocet_ok >= 2: b += 1 print(f'(a) P={a/celkem:.3f}') print(f'(b) P={b/celkem:.3f}')