AYTO

A Python library for calculating couple probabilities for the TV show Are You the One?.

Installation

pip install ayto

Usage

For a walkthrough of all of ayto's functionality, see the demo notebook.

from ayto import AYTO

guys = ["Albert", "Bill", "Carl"]
girls = ["Daisy", "Emily", "Faith"]

season = AYTO(guys, girls)

# "there are 6 possible scenarios"
print(f"there are {season.num_scenarios} possible scenarios")

# Al and Daisy go to the Truth Booth and get "NO MATCH"
scenarios_left = season.apply_truth_booth("Albert", "Daisy", False)

# "4 scenarios remain"
print(f"{scenarios_left} scenarios remain")

# A matchup ceremony with 1 beam
scenarios_left = season.apply_matchup_ceremony(
    [("Albert", "Emily"), ("Bill", "Daisy"), ("Carl", "Faith")], beams=1
)

# "2 scenarios remain"
print(f"{scenarios_left} scenarios remain")

print(season.probabilities)
#         Albert  Billy  Carl
# Daisy      0.0    0.5   0.5
# Emily      0.5    0.0   0.5
# Faith      0.5    0.5   0.0

season.save("my_season")

# ...later, to restore everything:
loaded_season = AYTO.load("my_season")
print(loaded_season.probabilities)
#         Albert  Billy  Carl
# Daisy      0.0    0.5   0.5
# Emily      0.5    0.0   0.5
# Faith      0.5    0.5   0.0