82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
import sys
|
|
import json
|
|
import pprint
|
|
|
|
pp = pp.Printer = pprint.PrettyPrinter(indent=2, compact=True)
|
|
|
|
def usagequit():
|
|
print("""Usage: python pydiceprob.py [k] [mode]
|
|
k between 6 and 99
|
|
modes: single, single-table, multi, multi-table
|
|
single prints out the probability of a single throw with a k-sided dice yielding k.
|
|
single-table prints out a table between 6 and k with the probabilities of a single throw yielding k.
|
|
multi prints out the probability of winning (assuming we're going first) in a game where two players take turns and the person who throws k first wins.
|
|
multi-table prints out the probability of winning for a range between 6 and k if you have the first throw.""")
|
|
quit()
|
|
|
|
def main():
|
|
global k
|
|
try:
|
|
if int(sys.argv[1]) >= 6:
|
|
k = int(sys.argv[1])
|
|
else:
|
|
usagequit()
|
|
except IndexError:
|
|
usagequit()
|
|
global mode
|
|
try:
|
|
mode = sys.argv[2]
|
|
except IndexError:
|
|
usagequit()
|
|
dispatch(mode, k)
|
|
|
|
def dispatch(mode, k):
|
|
modes = ["single", "multi", "single-table", "multi-table"]
|
|
if mode not in modes:
|
|
usagequit()
|
|
if mode == "single":
|
|
print(first_turn_probability(k))
|
|
elif mode == "multi" :
|
|
print(multi_turn_single(k))
|
|
elif mode == "single-table":
|
|
print(first_turn_protatbilities(k))
|
|
elif mode == "multi-table":
|
|
print(multi_turn_table(k))
|
|
|
|
def dice(k):
|
|
if k < 6:
|
|
print("Dice must be at least 6-sided")
|
|
elif k == 6:
|
|
dice_array = [6]
|
|
else:
|
|
dice_array = list(range(6, k))
|
|
|
|
def first_turn_probabilities(k):
|
|
result = {}
|
|
for k in range(6, int(k)+1):
|
|
result[k] = 1/k
|
|
return result
|
|
|
|
def first_turn_probability(k):
|
|
return first_turn_probabilities(k)[k]
|
|
|
|
def multi_turn_single(k):
|
|
p_win = 1 / k # Winning probability on any given throw
|
|
p_lose = (k-1) / k # Losing probability on any given throw
|
|
bob_wins_prob_sum = 0
|
|
r = p_lose**2
|
|
probability_win = p_win / (1 - r)
|
|
return probability_win
|
|
|
|
def multi_turn_table(k):
|
|
result = {}
|
|
for i in range(6, int(k+1)):
|
|
result[i] = multi_turn_single(i)
|
|
return result
|
|
|
|
def print(stuff):
|
|
pp.pprint(stuff)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|