In [1]:
from kmr_stationary import kmr_compute_stationary

In [2]:
a = kmr_compute_stationary(n=4, payoffs=[[4,0], [3,2]], epsilon=0.01) # payoffs = stug hunt game
"""
This  shows
1 states
2 transition matrix
3 stationary distribution
"""
print a


['10000', '01000', '00100', '00010', '00001']
[[ 0.995    0.005    0.       0.       0.     ]
 [ 0.24875  0.7475   0.00375  0.       0.     ]
 [ 0.       0.4975   0.5      0.0025   0.     ]
 [ 0.       0.       0.00375  0.7475   0.24875]
 [ 0.       0.       0.       0.005    0.995  ]]
[  9.75249963e-01   1.96030143e-02   1.47761415e-04   9.85076097e-05
   4.90075358e-03]

In [3]:
b = kmr_compute_stationary(n=8, payoffs=[[6,0,0], [5,7,5], [0,5,8]], epsilon=0.01) # payoffs = Young's game
"""
This function shows
1 states
2 transition matrix
3 stationary distribution
"""
print b


['1100000000', '0110000000', '1010000000', '0101000000', '0011000000', '1001000000', '0100100000', '0010100000', '0001100000', '1000100000', '0100010000', '0010010000', '0001010000', '0000110000', '1000010000', '0100001000', '0010001000', '0001001000', '0000101000', '0000011000', '1000001000', '0100000100', '0010000100', '0001000100', '0000100100', '0000010100', '0000001100', '1000000100', '0100000010', '0010000010', '0001000010', '0000100010', '0000010010', '0000001010', '0000000110', '1000000010', '0100000001', '0010000001', '0001000001', '0000100001', '0000010001', '0000001001', '0000000101', '0000000011', '1000000001']
[[  9.93333333e-01   3.33333333e-03   3.33333333e-03 ...,   0.00000000e+00
    0.00000000e+00   0.00000000e+00]
 [  1.24166667e-01   8.69583333e-01   4.16666667e-04 ...,   0.00000000e+00
    0.00000000e+00   0.00000000e+00]
 [  1.24166667e-01   4.16666667e-04   8.69583333e-01 ...,   0.00000000e+00
    0.00000000e+00   0.00000000e+00]
 ..., 
 [  0.00000000e+00   0.00000000e+00   0.00000000e+00 ...,   8.69583333e-01
    1.24166667e-01   0.00000000e+00]
 [  0.00000000e+00   0.00000000e+00   0.00000000e+00 ...,   3.33333333e-03
    9.93333333e-01   0.00000000e+00]
 [  0.00000000e+00   0.00000000e+00   0.00000000e+00 ...,   0.00000000e+00
    0.00000000e+00   9.93333333e-01]]
[  9.23036581e-03   2.47753131e-04   2.47836979e-04   5.82842144e-06
   2.88439112e-06   2.92856587e-06   6.01803471e-08   6.49545325e-08
   7.26451041e-09   3.11008561e-08   2.02729359e-09   3.01941578e-09
   5.33792126e-09   1.74265198e-11   8.36641561e-09   3.54658088e-08
   4.77738280e-09   3.57317768e-09   2.35669913e-11   3.90239548e-14
   1.97603340e-06   5.96434107e-06   5.50332154e-08   2.39395704e-09
   2.19452057e-11   7.17088860e-14   8.67269588e-17   2.95949280e-04
   5.91933399e-04   5.96347518e-06   3.44449597e-08   1.26285020e-10
   2.96831938e-13   4.30642300e-16   2.60564185e-15   2.51990257e-02
   2.51990552e-02   2.95964860e-04   1.98672133e-06   8.33810009e-09
   2.24080174e-11   3.76357781e-14   2.84480778e-15   1.01514624e-13
   9.38664257e-01]

In [ ]: