The 100 Prisoners Riddle

I found it hard to believe Veritasium's claims about the 100 Prisoners Riddle, so I wrote a small experiment to confirm that the proposed solution did in fact improve the odds so drastically.

The video is available here:

And here's the code I wrote to verify the solution works as expected, including a short description of the problem:

The output of the code shows clearly that the strategy works as claimed:

➜  prisoner_riddle python3 prisoner.py
Naive strategy gave 0.0% chance of success
Looping strategy gave 31.313000000000002% chance of success
➜  prisoner_riddle