xarray_einstats.tutorial.generate_mcmc_like_dataset

xarray_einstats.tutorial.generate_mcmc_like_dataset(seed=None)

Generate a Dataset with multiple variables, some with dimensions from mcmc sampling.

Parameters:

seedint or sequence of int, optional

The random seed used to initialize numpy.random.default_rng.

Notes

This function is not part of the public API and is designed for use in our documentation. In addition to generating the data, it also sets display_expand_data=False to avoid taking too much virtual space with examples.

Examples

The dataset generated is the following:

from xarray_einstats import tutorial
tutorial.generate_mcmc_like_dataset(3)

<xarray.Dataset>
Dimensions:  (plot_dim: 20, chain: 4, draw: 10, team: 6, match: 12)
Coordinates:
  * team     (team) <U1 'a' 'b' 'c' 'd' 'e' 'f'
  * chain    (chain) int64 0 1 2 3
  * draw     (draw) int64 0 1 2 3 4 5 6 7 8 9
Dimensions without coordinates: plot_dim, match
Data variables:
    x_plot   (plot_dim) float64 0.0 0.5263 1.053 1.579 ... 8.947 9.474 10.0
    mu       (chain, draw, team) float64 0.11 0.3897 1.4 ... 0.5328 0.3053
    sigma    (chain, draw) float64 2.385 0.5272 1.184 ... 0.1716 0.372 1.572
    score    (chain, draw, match) int64 2 2 0 0 2 2 1 0 1 ... 1 0 1 2 2 0 1 2 0

xarray.Dataset

• Dimensions:
  • plot_dim: 20
  • chain: 4
  • draw: 10
  • team: 6
  • match: 12
• Coordinates: (3)
  • team
    (team)
    <U1
    'a' 'b' 'c' 'd' 'e' 'f'

    array(['a', 'b', 'c', 'd', 'e', 'f'], dtype='<U1')

  • chain
    (chain)
    int64
    0 1 2 3

    array([0, 1, 2, 3])

  • draw
    (draw)
    int64
    0 1 2 3 4 5 6 7 8 9

    array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

• Data variables: (4)
  • x_plot
    (plot_dim)
    float64
    0.0 0.5263 1.053 ... 9.474 10.0

    array([ 0.        ,  0.52631579,  1.05263158,  1.57894737,  2.10526316,
            2.63157895,  3.15789474,  3.68421053,  4.21052632,  4.73684211,
            5.26315789,  5.78947368,  6.31578947,  6.84210526,  7.36842105,
            7.89473684,  8.42105263,  8.94736842,  9.47368421, 10.        ])

  • mu
    (chain, draw, team)
    float64
    0.11 0.3897 1.4 ... 0.5328 0.3053

    array([[[1.10014813e-01, 3.89656874e-01, 1.39954096e+00, 2.20014810e+00,
             3.43493819e-01, 2.57835523e-01],
            [4.34546514e-01, 1.02593270e-01, 1.00096157e+00, 6.63080394e-02,
             4.51260383e-01, 3.34763849e+00],
            [2.66531379e-01, 1.28534646e+00, 5.16016066e-01, 2.03425554e+00,
             9.07669713e-01, 1.48481113e+00],
            [2.89287612e-01, 1.77727224e-01, 1.79952899e-03, 1.16502494e+00,
             8.94422738e-01, 1.21619893e+00],
            [6.53590930e-01, 5.74610728e-01, 5.89804982e-01, 5.83932389e-01,
             4.13505747e-02, 2.15687275e+00],
            [1.41436886e+00, 6.55784053e-02, 1.45743898e+00, 3.06143496e+00,
             2.05667568e-01, 7.74694764e-01],
            [1.28074149e-01, 1.22973085e+00, 2.12729127e+00, 9.05993841e-01,
             1.35526536e+00, 1.15720544e+00],
            [7.58151304e-01, 1.77021215e+00, 3.88757660e-01, 2.04012840e+00,
             4.44009229e+00, 1.96581608e-01],
            [2.37972896e+00, 7.13865378e-01, 5.17181387e-01, 2.04374258e-01,
             3.12909791e+00, 1.25417645e-01],
            [1.46604013e+00, 6.12113869e-01, 2.51702650e+00, 1.15730361e+00,
             4.25422540e+00, 2.05697119e-01]],
    ...
           [[1.06632833e-03, 4.44743454e-01, 2.20822294e+00, 7.61873284e-01,
             7.83475799e-01, 1.93847897e+00],
            [3.79884039e+00, 2.43947566e+00, 1.33014755e+00, 2.79202624e-01,
             1.03162052e+00, 4.82291592e-01],
            [3.46867001e+00, 8.72982529e-01, 2.13031816e+00, 1.40251013e+00,
             7.44061073e-02, 1.79984111e+00],
            [1.68419335e+00, 2.16712699e+00, 1.17466741e+00, 2.14352945e+00,
             8.86728920e-01, 7.28975670e-01],
            [1.27978715e-01, 1.46906037e+00, 2.03623301e+00, 1.82234239e+00,
             2.92873124e-01, 7.41018952e-02],
            [3.05822380e+00, 1.05321050e+00, 1.07457773e+00, 1.04444461e+00,
             8.61747758e-01, 9.99543041e-02],
            [1.86365980e+00, 1.41440721e+00, 5.00290387e+00, 9.66444688e-03,
             1.31427377e+00, 2.32033342e-01],
            [5.50158338e-01, 7.28570620e-01, 4.71440080e-01, 2.04521208e+00,
             9.34517765e-01, 1.02014112e+00],
            [2.40458149e-01, 9.53811906e-01, 3.68343300e+00, 2.29933285e+00,
             2.47667666e-01, 5.46681963e-01],
            [1.99865931e+00, 3.14854775e-01, 4.91851809e-01, 7.30761472e-01,
             5.32769901e-01, 3.05313645e-01]]])

  • sigma
    (chain, draw)
    float64
    2.385 0.5272 1.184 ... 0.372 1.572

    array([[2.38493953, 0.52719228, 1.18372023, 0.12386704, 2.63289459,
            2.93133194, 1.86709274, 0.70532447, 0.5627929 , 2.48046136],
           [0.32759227, 0.08828297, 0.00603255, 3.00834605, 3.60777042,
            0.23472813, 0.28120151, 0.04637753, 1.40509805, 0.95953118],
           [2.96099052, 0.22385122, 0.31289612, 0.65856274, 1.19343632,
            0.26029851, 3.25030842, 0.48981837, 1.29680321, 0.70208348],
           [0.56545163, 1.71557953, 1.11905012, 3.07756236, 0.04574632,
            1.35081683, 0.65106143, 0.17161081, 0.3719661 , 1.57241203]])

  • score
    (chain, draw, match)
    int64
    2 2 0 0 2 2 1 0 ... 0 1 2 2 0 1 2 0

    array([[[2, 2, 0, 0, 2, 2, 1, 0, 1, 2, 0, 1],
            [1, 1, 1, 1, 1, 1, 2, 1, 0, 1, 0, 0],
            [1, 0, 0, 0, 3, 0, 0, 0, 0, 2, 0, 2],
            [0, 2, 2, 3, 0, 1, 2, 1, 1, 0, 1, 0],
            [0, 1, 0, 2, 0, 3, 2, 0, 0, 2, 1, 1],
            [1, 1, 0, 1, 0, 2, 0, 1, 0, 2, 0, 1],
            [0, 0, 0, 1, 1, 0, 0, 0, 1, 1, 2, 1],
            [1, 3, 1, 1, 2, 2, 1, 0, 1, 0, 1, 1],
            [0, 1, 0, 0, 0, 1, 2, 0, 1, 1, 3, 2],
            [1, 0, 1, 1, 3, 1, 2, 0, 0, 0, 3, 1]],
    
           [[1, 2, 1, 0, 1, 1, 0, 1, 0, 1, 0, 1],
            [1, 0, 4, 0, 1, 0, 1, 2, 2, 3, 1, 0],
            [1, 0, 0, 0, 4, 0, 2, 0, 1, 2, 1, 1],
            [0, 0, 0, 1, 0, 0, 2, 0, 0, 1, 2, 2],
            [1, 1, 1, 1, 0, 1, 1, 1, 0, 1, 0, 2],
            [0, 0, 1, 1, 0, 0, 2, 1, 0, 1, 2, 1],
            [1, 0, 0, 2, 2, 0, 0, 1, 0, 0, 1, 2],
            [2, 0, 1, 0, 0, 0, 2, 3, 0, 4, 1, 1],
            [0, 0, 0, 1, 2, 2, 0, 0, 1, 1, 1, 0],
    ...
            [1, 1, 2, 1, 0, 1, 1, 3, 0, 2, 1, 0],
            [2, 0, 0, 1, 0, 1, 0, 1, 0, 2, 3, 1],
            [1, 1, 1, 1, 1, 1, 1, 0, 1, 2, 2, 0],
            [0, 0, 0, 1, 2, 0, 0, 1, 1, 0, 3, 1],
            [1, 2, 1, 1, 1, 2, 3, 1, 1, 1, 0, 0],
            [0, 0, 0, 1, 0, 0, 1, 1, 3, 0, 1, 2],
            [0, 3, 2, 2, 2, 0, 0, 0, 0, 1, 2, 1],
            [4, 1, 1, 2, 0, 1, 0, 0, 0, 0, 2, 1],
            [1, 2, 2, 1, 0, 1, 3, 2, 2, 0, 0, 0]],
    
           [[3, 0, 2, 2, 0, 1, 1, 1, 2, 1, 1, 2],
            [1, 1, 1, 0, 0, 2, 1, 0, 1, 2, 0, 0],
            [0, 0, 1, 2, 2, 1, 0, 0, 0, 1, 1, 2],
            [0, 2, 1, 0, 0, 0, 1, 1, 2, 0, 2, 3],
            [1, 2, 0, 1, 3, 5, 0, 0, 1, 3, 0, 1],
            [2, 0, 0, 1, 3, 1, 1, 3, 0, 3, 0, 0],
            [3, 2, 2, 2, 0, 0, 1, 1, 0, 0, 0, 1],
            [2, 1, 1, 0, 2, 0, 2, 2, 1, 1, 2, 0],
            [2, 1, 1, 1, 1, 0, 2, 1, 0, 0, 0, 2],
            [2, 1, 0, 1, 0, 1, 2, 2, 0, 1, 2, 0]]])

• Attributes: (0)