arviz_base.explode_dataset_dims

arviz_base.explode_dataset_dims(ds, dim, labeller=None)

Explode dims of a dataset so each slice along them becomes its own variable.

Parameters:

dsxarray.Dataset

dimhashable or sequence of hashable

Dimension or dimensions along which slices to be stored as independent variables should be defined.

labellerlabeller, optional

Instance of a labeller class used to label the slices generated when exploding along dim. The method make_label_flat is used.

Returns:

xarray.Dataset

The dataset with all variables that have dim exploded into the respective slices as new variables.

Examples

In some cases, instead of theta as a (..., school) shape variable we’ll want independent variables for each slice:

from arviz_base import load_arviz_data, explode_dataset_dims
import xarray as xr

idata = load_arviz_data("centered_eight")
explode_dataset_dims(idata.posterior.dataset, "school")

<xarray.Dataset> Size: 164kB
Dimensions:                  (chain: 4, draw: 500)
Coordinates:
  * chain                    (chain) int64 32B 0 1 2 3
  * draw                     (draw) int64 4kB 0 1 2 3 4 ... 495 496 497 498 499
Data variables:
    mu                       (chain, draw) float64 16kB 7.872 3.385 ... 3.404
    theta[Choate]            (chain, draw) float64 16kB 12.32 11.29 ... 0.193
    theta[Deerfield]         (chain, draw) float64 16kB 9.905 9.129 ... 6.498
    theta[Phillips Andover]  (chain, draw) float64 16kB 14.95 3.139 ... -0.8944
    theta[Phillips Exeter]   (chain, draw) float64 16kB 11.01 9.433 ... 6.849
    theta[Hotchkiss]         (chain, draw) float64 16kB 5.58 7.812 ... 1.86
    theta[Lawrenceville]     (chain, draw) float64 16kB 16.9 2.393 ... 7.936
    theta[St. Paul's]        (chain, draw) float64 16kB 13.2 10.06 ... 6.762
    theta[Mt. Hermon]        (chain, draw) float64 16kB 15.06 6.177 ... 1.295
    tau                      (chain, draw) float64 16kB 4.726 3.909 ... 4.461

xarray.Dataset

• Dimensions:
  • chain: 4
  • draw: 500
• Coordinates: (2)
  • chain
    (chain)
    int64
    0 1 2 3

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

  • draw
    (draw)
    int64
    0 1 2 3 4 5 ... 495 496 497 498 499

    array([  0,   1,   2, ..., 497, 498, 499], shape=(500,))

• Data variables: (10)
  • mu
    (chain, draw)
    float64
    7.872 3.385 9.1 ... 3.486 3.404

    array([[7.87179637, 3.38455431, 9.10047569, ..., 3.65680422, 5.42727938,
            2.73588293],
           [4.31581657, 4.59770112, 6.21376164, ..., 4.84102392, 5.94610513,
            4.66643827],
           [4.52760336, 5.86262758, 5.91070276, ..., 2.87106698, 4.09553646,
            1.77571755],
           [7.17950438, 6.53969896, 7.48361981, ..., 1.76673325, 3.48611194,
            3.40446391]], shape=(4, 500))

  • theta[Choate]
    (chain, draw)
    float64
    12.32 11.29 5.709 ... 4.183 0.193

    array([[12.32068558, 11.28562322,  5.70850575, ..., -0.21382811,
            10.40250133,  6.66130764],
           [ 5.59663866,  4.96540992,  5.89670043, ...,  4.15384252,
             7.03060429,  4.884931  ],
           [11.81416982,  4.74346536,  8.08257446, ...,  4.71860289,
             5.67765139,  1.62544668],
           [ 8.34450834,  8.93115045,  9.62317435, ...,  3.53251469,
             4.1827505 ,  0.19295578]], shape=(4, 500))

  • theta[Deerfield]
    (chain, draw)
    float64
    9.905 9.129 5.758 ... 7.554 6.498

    array([[ 9.90536689,  9.12932369,  5.7579324 , ...,  1.35515489,
             6.90740688,  7.41376703],
           [ 6.95960523,  1.77935809,  9.47253003, ...,  4.18672334,
             6.21446547,  3.9522469 ],
           [-0.04047735, 10.17568393,  7.225948  , ...,  6.3386488 ,
             1.67910823,  2.50328085],
           [ 5.39085524,  6.85296853,  5.10464403, ...,  2.00890089,
             7.55425055,  6.4984278 ]], shape=(4, 500))

  • theta[Phillips Andover]
    (chain, draw)
    float64
    14.95 3.139 10.94 ... 4.456 -0.8944

    array([[14.9516155 ,  3.13926327, 10.94458539, ...,  6.98268796,
            -4.96413775, -9.32260183],
           [ 3.2412923 ,  6.82078866,  5.29354995, ...,  2.65485994,
             8.69054158,  3.10464827],
           [ 1.56282554,  4.26541319,  6.9902812 , ...,  7.67278512,
             2.97637036,  0.77298319],
           [ 4.71615276, 10.07129962,  4.99209455, ...,  0.51080602,
             4.45603444, -0.89442402]], shape=(4, 500))

  • theta[Phillips Exeter]
    (chain, draw)
    float64
    11.01 9.433 5.895 ... 3.301 6.849

    array([[11.01148494,  9.43321074,  5.89543593, ...,  3.7168145 ,
             3.13584364,  2.69191633],
           [ 3.22518066,  6.09657359,  4.6322701 , ...,  3.35951658,
             7.9545155 ,  5.12806495],
           [ 3.68632412,  5.93742059,  3.20675887, ...,  4.96867135,
            -0.87695432,  4.28218277],
           [10.76407817,  1.28831383, 12.44181403, ...,  0.8321855 ,
             3.30083344,  6.84901981]], shape=(4, 500))

  • theta[Hotchkiss]
    (chain, draw)
    float64
    5.58 7.812 9.993 ... 1.563 1.86

    array([[ 5.57960159,  7.81151556,  9.99298421, ...,  5.32446227,
            -2.22430206, -0.5023312 ],
           [ 3.49991232,  6.35851737,  2.15775748, ...,  4.38305018,
             5.63650668,  5.87736538],
           [ 6.9838363 ,  2.29043625,  1.37793663, ...,  3.76781835,
             4.8200791 ,  1.10303418],
           [-9.08554302, 12.55040351, -0.75013665, ...,  2.64768722,
             1.56330704,  1.85974681]], shape=(4, 500))

  • theta[Lawrenceville]
    (chain, draw)
    float64
    16.9 2.393 8.143 ... 1.529 7.936

    array([[16.90179529,  2.39308809,  8.14332707, ...,  6.96589044,
            -2.83504069, -4.25487418],
           [ 5.5731038 ,  4.63345858,  5.65256064, ...,  2.83710048,
             7.19175218,  5.25116805],
           [ 2.72540024,  7.42048044,  8.00492008, ...,  1.85496043,
             1.61054881,  1.47752619],
           [ 7.87559218,  0.11111666, 11.45121967, ...,  4.70724881,
             1.5289581 ,  7.93646013]], shape=(4, 500))

  • theta[St. Paul's]
    (chain, draw)
    float64
    13.2 10.06 7.605 ... 1.096 6.762

    array([[13.19805933, 10.05522282,  7.60475341, ...,  4.93020022,
             5.3910607 ,  7.56657019],
           [ 7.14885429,  3.28916128,  3.30221324, ...,  6.16798642,
             4.77813768,  5.68793544],
           [ 0.31808711, 11.27171121,  7.82910303, ...,  7.29157333,
             1.28061498,  2.36275624],
           [12.46813994,  7.01397138,  9.45569534, ...,  3.07331446,
             1.09609826,  6.76245459]], shape=(4, 500))

  • theta[Mt. Hermon]
    (chain, draw)
    float64
    15.06 6.177 8.768 ... 8.452 1.295

    array([[15.06136584,  6.17672421,  8.76764689, ...,  3.05860034,
             6.38123908,  9.98761823],
           [ 3.79090016,  5.93731077,  7.16213323, ...,  7.09633148,
             4.35711209,  6.77777189],
           [ 0.66130401,  7.12114024,  3.11374388, ..., -0.10026455,
             7.62765755, -2.96799418],
           [12.60779651,  5.13629724,  8.85937825, ..., -2.62306892,
             8.45228161,  1.2950506 ]], shape=(4, 500))

  • tau
    (chain, draw)
    float64
    4.726 3.909 4.844 ... 2.932 4.461

    array([[4.72574006, 3.90899361, 4.8440252 , ..., 1.89383828, 5.92006245,
            4.32589567],
           [1.97083011, 2.04902913, 2.12376496, ..., 2.17459002, 1.32755081,
            1.21199466],
           [3.50127689, 2.89324317, 4.27328591, ..., 4.08978031, 2.72016996,
            1.91701121],
           [6.07325982, 3.77186702, 3.17053747, ..., 2.74060666, 2.93237926,
            4.46124596]], shape=(4, 500))

• Indexes: (2)
  • 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,
           ...
           490, 491, 492, 493, 494, 495, 496, 497, 498, 499],
          dtype='int64', name='draw', length=500))

• Attributes: (0)