import matplotlib.pyplot as plt
import numpy as np

# Задача 3: геометрическое распределение
p3 = 0.9
q3 = 1 - p3
xi3 = list(range(1, 11))  # ограничим до 10 вопросов
P3 = [p3 ** (k - 1) * q3 for k in xi3]
F3 = np.cumsum(P3)

plt.figure(figsize=(7, 5))
plt.bar(xi3, P3, color="lightgreen", alpha=0.7, label="Вероятности")
plt.step(xi3, F3, where="post", color="orange", label="Функция распределения")
plt.scatter(xi3, F3, color="red")
plt.title("Задача 3: геометрическое распределение")
plt.xlabel("xi")
plt.ylabel("P / F")
plt.xticks(xi3)
plt.legend()
plt.grid(axis="y", linestyle="--", alpha=0.6)
plt.tight_layout()
plt.savefig("3.png")