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> Size: 6kB
Dimensions:  (plot_dim: 20, chain: 4, draw: 10, team: 6, match: 12)
Coordinates:
  * team     (team) <U1 24B 'a' 'b' 'c' 'd' 'e' 'f'
  * chain    (chain) int64 32B 0 1 2 3
  * draw     (draw) int64 80B 0 1 2 3 4 5 6 7 8 9
Dimensions without coordinates: plot_dim, match
Data variables:
    x_plot   (plot_dim) float64 160B 0.0 0.5263 1.053 1.579 ... 8.947 9.474 10.0
    mu       (chain, draw, team) float64 2kB 0.11 0.3897 1.4 ... 0.5328 0.3053
    sigma    (chain, draw) float64 320B 2.385 0.5272 1.184 ... 0.372 1.572
    score    (chain, draw, match) int64 4kB 2 2 0 0 2 2 1 0 ... 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]]])

• Indexes: (3)
  • team
    PandasIndex

    PandasIndex(Index(['a', 'b', 'c', 'd', 'e', 'f'], dtype='object', name='team'))

  • chain
    PandasIndex

    PandasIndex(Index([0, 1, 2, 3], dtype='int64', name='chain'))

  • draw
    PandasIndex

    PandasIndex(Index([0, 1, 2, 3, 4, 5, 6, 7, 8, 9], dtype='int64', name='draw'))

• Attributes: (0)